leetcode265. 粉刷房子 II（Python3、c++）

leetcode265. 粉刷房子 II

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

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

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

注意：

所有花费均为正整数。

示例：

输入: [[1,5,3],[2,9,4]]
输出: 5
解释: 将 0 号房子粉刷成 0 号颜色，1 号房子粉刷成 2 号颜色。最少花费: 1 + 4 = 5; 
     或者将 0 号房子粉刷成 2 号颜色，1 号房子粉刷成 0 号颜色。最少花费: 3 + 2 = 5.

进阶：

您能否在 O(nk) 的时间复杂度下解决此问题？

---

方法一：动态规划（未优化）

思路：

本题与粉刷房子①类似，只不过从3种颜色变成了k种颜色。

因此，我们定义二维dp数组，从第一个房子开始涂色，**dp(i,j)表示房子i涂成颜色j，此时将0-i房子都涂完色的最小花费。**那么，我们的答案应该为dp[n-1][:]中的最小值。

下面我们考虑状态转移方程：

当我们涂到房子i的时候，我们需要考虑房子i-1的颜色。此时若将房子i涂成颜色j，**那么此时最小花费应该为dp(i-1,c)的最小值+costs(i,j)，这里c不等于j。即，将前i-1房子涂成不是j颜色中的k-1个最小花费中的最小值，加上将i涂成j的花费。**即：
$$dp[i][j]=min(dp[i-1][c])+cost[i][j]c\ne j$$
我们的初始值很容易，即dp(0,i) = cost(0,i)即可。

我们直接遍历颜色c，时间复杂度为O(k)，再加上填写dp时间复杂度为O(nk)，因此总的时间复杂度为O(nk^2)

代码：

Python3：

class Solution:
    def minCostII(self, costs: List[List[int]]) -> int:
        n = len(costs)
        if not n:
            return 0
        k = len(costs[0])
        # 从i=0房子开始涂色，dp[i][j]表示房子i涂成j颜色的情况下，0-i房子涂色完的最小花费
        dp