动态规划——粉刷房子

题目链接

leetcode在线oj题——粉刷房子

题目描述

假如有一排房子，共 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

题目提示

• costs.length == n
• costs[i].length == 3
• 1 <= n <= 100
• 1 <= costs[i][j] <= 20

解题思路

由于这道题有三个不同的颜色，而题目中使用cost[i][0]代表第i个房子涂成红色的花费，1代表蓝色，2代表绿色，因此我们的dp数组也可以使用0，1，2分别代表红蓝绿

如果从前往后思考，前面一个房子的颜色不确定，因此不知道后一个房子是什么情况，因此我们可以从后往前思考，确定了第i个房子是红色，考虑涂到当前位置需要多少花销

当第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][1], dp[i - 1][0]) + costs[i][2];

最终，直接返回最后一个位置是红蓝绿三种情况中最小花费的即可

完整代码

class Solution {
    public int minCost(int[][] costs) {
        int m = costs.length;
        int n = costs[0].length;
        int[][] dp = new int[m][n];
        //0:red 1:blue 2:green
        dp[0][0] = costs[0][0];
        dp[0][1] = costs[0][1];
        dp[0][2] = costs[0][2];
        for (int i = 1; i < m; 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][1], dp[i - 1][0]) + costs[i][2];
        }
        return Math.min(Math.min(dp[m - 1][0], dp[m - 1][1]), dp[m - 1][2]);
    }
}