【LeetCode256】粉刷房子

1.题目描述

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

2.解法

解法一:动态规划
class Solution:
    def minCost(self, costs: List[List[int]]) -> int:
        import sys
        pre0 = 0
        pre1 = 0 
        pre2 = 0
        for cost in costs:
            f0 = cost[0] + min(pre1, pre2)
            f1 = cost[1] + min(pre0, pre2)
            f2 = cost[2] + min(pre0, pre1)
            pre0, pre1, pre2 = f0, f1, f2
        return min(pre0, pre1, pre2)
经验分享 程序员 微信小程序 职场和发展