题目描述
假如有一排房子,共 n 个,每个房子可以被粉刷成红色、蓝色或者绿色这三种颜色中的一种,你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。 当然,因为市场上不同颜色油漆的价格不同,所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 3 的正整数矩阵 costs 来表示的。
例如,costs[0][0] 表示第 0 号房子粉刷成红色的成本花费;costs[1][2] 表示第 1 号房子粉刷成绿色的花费,以此类推。 请计算出粉刷完所有房子最少的花费成本。
动态规划原理
动态规划(Dynamic Programming, DP)是一种通过将原问题分解为相对简单的子问题的方式求解复杂问题的方法。这些子问题通常是原问题的较小实例,并且它们的解被保存起来以便后续使用,从而避免了重复计算。动态规划通常用于求解具有重叠子问题和最优子结构性质的问题。
解题步骤
- 定义状态:
- 在这个问题中,我们定义
dp[i][j]
为将前i
个房子粉刷完毕,并且第i
个房子粉刷成颜色j
的最小成本。
- 在这个问题中,我们定义
- 状态转移方程:
- 要计算
dp[i][j]
,我们需要考虑前一个房子i-1
的颜色。由于相邻房子颜色不能相同,因此我们需要从dp[i-1][k]
(其中k
不等于j
)中选择最小值,并加上当前房子粉刷成颜色j
的成本costs[i][j]
。 - 状态转移方程为:
dp[i][j] = min(dp[i-1][k] + costs[i][j])
,其中k
遍历0, 1, 2
且k != j
。
- 要计算
- 初始化:
- 对于第一个房子(
i = 0
),我们可以直接选择任意颜色,因此dp[0][j] = costs[0][j]
,其中j = 0, 1, 2
。
- 对于第一个房子(
- 计算顺序:
- 我们需要按照房子的顺序(从第一个房子到最后一个房子)来填充
dp
数组。
- 我们需要按照房子的顺序(从第一个房子到最后一个房子)来填充
- 结果:
- 最终答案将是
dp[n-1][j]
中的最小值,其中j = 0, 1, 2
,表示粉刷完所有房子后的最小成本。
- 最终答案将是
代码如下:
class Solution {
public:
int minCost(vector<vector<int>>& costs) {
int n = costs.size();
vector<vector<int>> dp(n, vector<int>(3, 0));
// 初始化第一个房子的成本
for (int i = 0; i < 3; i++) {
dp[0][i] = costs[0][i];
}
// 填充dp数组
for (int i = 1; i < n; i++) {
for (int j = 0; j < 3; j++) {
if (j == 0)
dp[i][j] =
min(dp[i - 1][j + 1], dp[i - 1][j + 2]) + costs[i][j];
else if (j == 1)
dp[i][j] =
min(dp[i - 1][j - 1], dp[i - 1][j + 1]) + costs[i][j];
else if (j == 2)
dp[i][j] =
min(dp[i - 1][j - 2], dp[i - 1][j - 1]) + costs[i][j];
}
}
// 返回最小成本
return min(dp[n - 1][0], min(dp[n - 1][1], dp[n - 1][2]));
}
};