假如有一排房子,共 n 个,每个房子可以被粉刷成 k 种颜色中的一种,你需要粉刷所有的房子并且使其相邻的两个房子颜色不能相同。
当然,因为市场上不同颜色油漆的价格不同,所以房子粉刷成不同颜色的花费成本也是不同的。每个房子粉刷成不同颜色的花费是以一个 n x 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) 的时间复杂度下解决此问题?
思路:
跟LeetCode-Python-256. 粉刷房子基本相同……
class Solution(object):
def minCostII(self, costs):
"""
:type costs: List[List[int]]
:rtype: int
"""
if not costs or not costs[0]:
return 0
dp = costs
def GetMin(idx, k):
Min = max(costs[idx])
for i, cost in enumerate(costs[idx]):
if i == k:
continue
Min = min(Min, cost)
return Min
for i in range(1, len(costs)):
for k in range(len(costs[i])):
dp[i][k] += GetMin(i - 1, k) #从上一行下标不为k的元素里找最小的那个
return min(dp[-1])