问题:有一排N栋房子,每栋房子要油漆成红、蓝、绿三种颜色,第i栋房子油漆成红、蓝、绿的花费分别为cost[i][0]、cost[i][1]、cost[i][2]
要求:相邻的房子颜色需不同
求油漆好房子的最小花费
代码及注释如下:
def Pint_huose(cost):
#N是房子的个数,K是油漆的颜色种数
N,K = len(cost),len(cost[0])
#f[i][j]表示油漆前i个房子并且房子i-1是颜色j的最小花费,因为求最小,所以都先设置成大数
f = [[sys.maxsize for x in range(K)] for y in range(N+1)]
#初始条件
for z in range(K):
f[0][z] = 0
for i in range(1,N+1):
#从前1栋房子到前N栋房子
for j in range(K):
#j是房子i-1的颜色
for k in range(K):
#k是房子i-2使用的颜色
if j == k:
#如果房子i-1的颜色和前面的房子同色,则略过
continue
if f[i-1][k] + cost[i-1][j] < f[i][j]:
#f[i-1][k]表示油漆前i-1栋房子并且房子i-1使用颜色k的最小花费
#cost[i-1][j]表示房子i-1用颜色j油漆的花费
#如果房子油漆前i-1个房子且房子i-2用颜色k,房子i-1用