[Python]lintcode 515. 房屋染色 中文English 这里有n个房子在一列直线上,你需要设计一种染色方案使得相邻的房屋颜色不同,并且费用最小

lintcode 515. 房屋染色

中文English
这里有n个房子在一列直线上，现在我们需要给房屋染色，分别有红色蓝色和绿色。每个房屋染不同的颜色费用也不同，你需要设计一种染色方案使得相邻的房屋颜色不同，并且费用最小，返回最小的费用。费用通过一个nx3 的矩阵给出，比如cost[0][0]表示房屋0染红色的费用，cost[1][2]表示房屋1染绿色的费用。

代码

def minCost(self, costs):
    # write your code here
    n=len(costs)
    if(n==0):
        return 0
    dp=[[0]*3 for i in range(n+1)] 
    dp[0][0]=dp[0][1]=dp[0][2]=0
    for i in range(1,n+1,1):
        for j in range(0,3,1):
            dp[i][j]=float('inf')
            if j==0:
                dp[i][j] = min(dp[i - 1][1]+ costs[i-1][j], dp[i - 1][2]+ costs[i-1][j])
            if j==1:
                dp[i][j] = min(dp[i - 1][0]+ costs[i-1][j], dp[i - 1][2]+ costs[i-1][j]) 
            if j==2:
                dp[i][j] = min(dp[i - 1][0]+ costs[i-1][j], dp[i - 1][1]+ costs[i-1][j])            
                
    return min(dp[n][0], min(dp[n][1], dp[n][2]))

想法

1.前i栋房子中在最后面的是第i-1栋
2.前面代码dp[i][j]指的是前i栋以及第i-1栋的颜色
costs[i][j]表示房屋i染某一颜色的费用
所以前i栋房子的消费，应该前i-1栋房子的消费加上第i-1栋房子的消费