lintcode python 代码 514 栅栏染色

http://www.th7.cn/Program/Python/201706/1186615.shtml

我们有一个栅栏，它有n个柱子，现在要给柱子染色，有k种颜色可以染。
必须保证任意两个相邻的柱子颜色不同，求有多少种染色方案。
思路：第一个柱子有k中方案 number[1]=k
第二个柱子有k中方案 number[2]=k*k
第三个柱子： 1.和第二个柱子同色 和第一个不同色 number[2]=k*（k-1）
2.和第二个柱子不同色 number[2]=k*k*（k-1）
第n个柱子number[n]=record[n- 2] * (k - 1) + record[n-1] * (k - 1)

class Solution:
    # @param {int} n non-negative integer, n posts
    # @param {int} k non-negative integer, k colors
    # @return {int} an integer, the total number of ways
    def numWays(self, n, k):
        # Write your code here
        if n==1 or n==2:
            return pow(k,n)
        if k<=1:
            return 0
        result=[0]* n #初始化放每个柱子的染色方案
        result[0]=k
        result[1]=k*k
        for i in range(2,n):
            result[i]=result[i-1]*(k-1)+result[i-2]*(k-1)
        return result[-1]  #输出n个柱子的染色情况

上面是按着那个网址

补充思路：
第三个柱子

• 与第二个相同与第一个不相同：k*(k-1)1 =第一个柱子情况（k-1）
  ps:如果与第二个相同，那么第一个肯定不相同，相邻最多出现两个相同。
• 与第二个不相同：
  –>与第一个相同：k*(k-1)*1 ps:第一个与第二个也不可能相同
  –>与第一个不相同:k*（k-1）(k-2)+k*1(k-1) ps:第二个和第一个相同+不相同的情况
  合计为k*(k-1)1+k（k-1）(k-2)+k*1(k-1)=k*(k-1)(1+k-2+1)=k*(k-1)k=第二个柱子情况（k-1）