欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

DP-LeetCode256. 粉刷房子

程序员文章站 2022-03-24 20:52:51
...

1、题目描述

https://www.lintcode.com/problem/paint-house/description

DP-LeetCode256. 粉刷房子

注意:所有花费均为正整数。

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

2、代码详解

转移方程

DP-LeetCode256. 粉刷房子

DP-LeetCode256. 粉刷房子

初始条件和边界情况

DP-LeetCode256. 粉刷房子

计算顺序

DP-LeetCode256. 粉刷房子

class Solution(object):
    def minCost(self, costs):
        """
        :type costs: List[List[int]]
        :rtype: int
        """
        MAX = float('inf')
        n = len(costs)
        if n == 0:
            return 0

        dp = [[MAX]*3 for _ in range(n+1)]  # n+1 * 3 (从0-n,表示前n个房子的累计)
        # 初始条件
        dp[0][0], dp[0][1], dp[0][2] = 0, 0, 0  # 前0个房子,没有房子

        for i in range(1, n+1):  # 此处i-1个房子,即当前的房子

            # 第i-1个房子的颜色j(当前房子)
            for j in range(3):

                # 第i-2个房子的颜色k
                for k in range(3):
                    if j != k:  # 邻居不撞色
                        # 前一个房子染k色 + 当前房子染j色的cost(dp和cost下标不是对齐的)
                        dp[i][j] = min(dp[i][j], dp[i-1][k] + costs[i-1][j])

        return min(dp[n][0], dp[n][1], dp[n][2])

costs = [[14,2,11],
         [11,4,5],
         [14,3,10]]  # 2+5+3=10
costs1 = [[17,2,17],
          [16,16,5],
          [14,3,19]]  # 2+5+3=10
s = Solution()
print(s.minCost(costs1))

其他优质解法

https://blog.csdn.net/qq_32424059/article/details/90414481

 

 

 

 

相关标签: 动态规划