leetcode256. 粉刷房子

忘是亡心i 2023-07-05 10:29 18阅读 0赞

假如有一排房子，共 n 个，每个房子可以被粉刷成红色、蓝色或者绿色这三种颜色中的一种，你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。

当然，因为市场上不同颜色油漆的价格不同，所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 3 的矩阵来表示的。

例如，costs\[0\]\[0\] 表示第 0 号房子粉刷成红色的成本花费；costs\[1\]\[2\] 表示第 1 号房子粉刷成绿色的花费，以此类推。请你计算出粉刷完所有房子最少的花费成本。

注意：

所有花费均为正整数。

示例：

输入: \[\[17,2,17\],\[16,16,5\],\[14,3,19\]\]  
输出: 10  
解释: 将 0 号房子粉刷成蓝色，1 号房子粉刷成绿色，2 号房子粉刷成蓝色。  
最少花费: 2 + 5 + 3 = 10。

思路：dp\[i\]\[0,1,2\]分别代表第i个格子必须用红、蓝、绿刷的最优解。则dp\[i\]\[0\]的前一个格子可以是蓝或绿，也就是dp\[i-1\]\[1和dp\[i-1\]\[2\]两种情况，取最优解加上cost\[i\]\[0\]即可。另两种类似。

dp\[i\]只和dp\[i-1\]有关，所以用几个变量记录即可。

class Solution {
        public int minCost(int[][] costs) {
            if(costs == null || costs.length == 0) return 0;
            int r_cost = costs[0][0];
            int b_cost = costs[0][1];
            int g_cost = costs[0][2];
            for(int i = 1; i < costs.length; i++){
                int r_tmp = r_cost;
                int b_tmp = b_cost;
                int g_tmp = g_cost;
                r_cost = costs[i][0] + Math.min(b_tmp, g_tmp);
                b_cost = costs[i][1] + Math.min(r_tmp, g_tmp);
                g_cost = costs[i][2] + Math.min(r_tmp, b_tmp);
            }
            return  Math.min(Math.min(r_cost, b_cost),g_cost);
        }
    }