当前位置: 首页 > 工具软件 > 蓝房子 > 使用案例 >

leetcode256. 粉刷房子

唐修能
2023-12-01

假如有一排房子,共 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);
    }
}

 

 类似资料: