Question

剑指 Offer II 091. 粉刷房子

题目描述

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

 

注意：本题与主站 256 题相同：https://leetcode.cn/problems/paint-house/

解法

方法一

class Solution:
  def minCost(self, costs: List[List[int]]) -> int:
    r, g, b = 0, 0, 0
    for cost in costs:
      _r, _g, _b = r, g, b
      r = min(_g, _b) + cost[0]
      g = min(_r, _b) + cost[1]
      b = min(_r, _g) + cost[2]
    return min(r, g, b)

class Solution {
  public int minCost(int[][] costs) {
    int r = 0, g = 0, b = 0;
    for (int[] cost : costs) {
      int _r = r, _g = g, _b = b;
      r = Math.min(_g, _b) + cost[0];
      g = Math.min(_r, _b) + cost[1];
      b = Math.min(_r, _g) + cost[2];
    }
    return Math.min(r, Math.min(g, b));
  }
}

class Solution {
public:
  int minCost(vector<vector<int>>& costs) {
    int r = 0, g = 0, b = 0;
    for (auto& cost : costs) {
      int _r = r, _g = g, _b = b;
      r = min(_g, _b) + cost[0];
      g = min(_r, _b) + cost[1];
      b = min(_r, _g) + cost[2];
    }
    return min(r, min(g, b));
  }
};

func minCost(costs [][]int) int {
  r, g, b := 0, 0, 0
  for _, cost := range costs {
    _r, _g, _b := r, g, b
    r = min(_g, _b) + cost[0]
    g = min(_r, _b) + cost[1]
    b = min(_r, _g) + cost[2]
  }
  return min(r, min(g, b))
}

function minCost(costs: number[][]): number {
  let [r, g, b] = [0, 0, 0];
  for (const [_r, _g, _b] of costs) {
    [r, g, b] = [_r + Math.min(g, b), _g + Math.min(r, b), _b + Math.min(r, g)];
  }
  return Math.min(r, g, b);
}

impl Solution {
  pub fn min_cost(costs: Vec<Vec<i32>>) -> i32 {
    let mut dp = [0, 0, 0];
    for cost in costs.iter() {
      dp = [
        cost[0] + dp[1].min(dp[2]),
        cost[1] + dp[0].min(dp[2]),
        cost[2] + dp[0].min(dp[1]),
      ];
    }
    *dp.iter().min().unwrap()
  }
}