题目如下
假如有一排房子,共 n 个,每个房子可以被粉刷成红色、蓝色或者绿色这三种颜色中的一种,你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。
当然,因为市场上不同颜色油漆的价格不同,所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 3 的正整数矩阵 costs 来表示的。
例如,costs[0][0] 表示第 0 号房子粉刷成红色的成本花费;costs[1][2] 表示第 1 号房子粉刷成绿色的花费,以此类推。
请计算出粉刷完所有房子最少的花费成本。
示例 1:
输入: costs = [[17,2,17],[16,16,5],[14,3,19]]
输出: 10
解释: 将 0 号房子粉刷成蓝色,1 号房子粉刷成绿色,2 号房子粉刷成蓝色。
最少花费: 2 + 5 + 3 = 10。
示例 2:
输入: costs = [[7,6,2]]
输出: 2
解题思路
每一步需要粉刷一幢房子,粉刷所有房屋需要 n 步,每一步又面临粉刷成红绿蓝三色的选择,最终需要计算最少成本,所以此题可使用动态规划求解。
设 r(i) 、g(i) 和 b(i) 分别表示将标号为 i 的房屋刷成红色、绿色和蓝色时粉刷标号从 0 到 i 的 i + 1 幢房子的最小成本。同时因为相邻的房屋不能涂成同一个颜色,所以可以得到状态转移方程为
完整代码如下,时间复杂度为 O(n),空间复杂度为 O(1)。
class Solution
{
public:
int minCost(vector<vector<int>>& costs)
{
vector<vector<int>>dp(2, vector<int>(3, 0));
dp[0] = costs[0];
for (int i = 1; i < costs.size(); ++i)
{
for (int j = 0; j < 3; ++j)
{
int prev1 = dp[(i - 1) % 2][(j + 1) % 3];
int prev2 = dp[(i - 1) % 2][(j + 2) % 3];
dp[i % 2][j] = min(prev1, prev2) + costs[i][j];
}
}
int last = (costs.size() - 1) % 2;
return min(min(dp[last][0], dp[last][1]), dp[last][2]);
}
};
class Solution
{
public:
int minCost(vector<vector<int>>& vec)
{
int[][] dp = new int[costs.length][3];
//第一件房子
for (int i = 0; i < 3; i++)
{
dp[0][i] = costs[0][i];
}
for (int i = 1; i < costs.length; i++)
{
//状态转换
dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + costs[i][0];
dp[i][1] = Math.min(dp[i - 1][0], dp[i - 1][2]) + costs[i][1];
dp[i][2] = Math.min(dp[i - 1][0], dp[i - 1][1]) + costs[i][2];
}
return Math.min(dp[costs.length - 1][0],
Math.min(dp[costs.length - 1][1], dp[costs.length - 1][2]));
}
};