LeetCode--265. 粉刷房子Ⅱ(动态规划)

粉刷房子Ⅱ(动态规划)

1. 题目描述

难度:困难
在这里插入图片描述

2. 题目分析

这道题目是256. 粉刷房子的进阶题,其实解题思路和256题是一样的:

  • 动态规划
    256题颜色是3种,但是本题的颜色变为了k种,但是整体的思路还是一样的,我们申请一个二维数组dp[n][k]来存放每个房子的每个颜色的最低花费,时间复杂度为O(nk)。

3. C语言实现

代码如下:

// 返回二者最小值
int Min(int a, int b){
    return a<b?a:b;
}
// 返回一个数组中的下标不是index的最小元素
int MinArray(int *cost, int index, int lens){
    int min = index==0?cost[1]:cost[0];
    for(int i = 0; i < lens; i++){
        if(i != index){
            min = Min(min, cost[i]);
        }
    }
    return min;
}
int minCostII(int** costs, int costsSize, int* costsColSize){
    int n = costsSize;
    int k = costsColSize[0];
    int res;
    if(n == 0) return 0;
    if(k == 1) return costs[0][0];
    int** dp = (int **)malloc(sizeof(int)*(n*k));
    for(int i = 0; i < n; i++) dp[i] = (int *)malloc(sizeof(int)*k);
    for(int i = 0; i < k; i++) dp[0][i] = costs[0][i];
    for(int i = 1; i < n; i++){
        for(int j = 0; j < k; j++){
            dp[i][j] = MinArray(dp[i-1], j, k) + costs[i][j];
        }
    }
    res = MinArray(dp[n-1], -1, k);
    free(dp);

    return res;
}

运行结果为:
在这里插入图片描述

发布了175 篇原创文章 · 获赞 237 · 访问量 12万+

猜你喜欢

转载自blog.csdn.net/qq_42580947/article/details/105294599
今日推荐