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;
}
运行结果为: