剑指 Offer II 091. 粉刷房子
题意:n个房子排成一排,用红绿蓝三种颜色给房子涂色,要求相邻房子不能涂相同的颜色。costs是一个n行3列的数组,costs[i][j]表示第i个房子涂j种颜色的花费,求涂完所有房子后所需的最小花费。
思路:动态规划。f[i][j]表示涂完1~i个房子所需的最小花费,对于第i个房子,它有3种选择,每种不同的选择取决于前一个房子的涂色情况。可列出下面的动态方程组:
0、1、2分别代表三种不同的颜色:
f[i][0] = min(f[i - 1][1], f[i - 1][2]) + costs[i][0];
f[i][1] = min(f[i - 1][0], f[i - 1][2]) + costs[i][1];
f[i][2] = min(f[i - 1][0], f[i - 1][1]) + costs[i][2];
class Solution {
public int minCost(int[][] costs) {
int n = costs.length;
int m = 3;
int[][] f = new int[n][m];
for(int i = 0; i < 3; i++) {
f[0][i] = costs[0][i];
}
for(int i = 1; i < n; i++) {
for(int j = 0; j < m; j++) {
if(j == 0) {
f[i][j] = Math.min(f[i - 1][1], f[i - 1][2]) + costs[i][j];
}
else if(j == 1) {
f[i][j] = Math.min(f[i - 1][2], f[i - 1][0]) + costs[i][j];
} else {
f[i][j] = Math.min(f[i - 1][0], f[i - 1][1]) + costs[i][j];
}
}
}
int ans = 0x3f3f3f;
for(int i = 0; i < 3; i++) {
ans = Math.min(ans, f[n - 1][i]);
}
return ans;
}
}