栅栏染色

栅栏染色 

•  描述
•  笔记
•  数据
•  评测

我们有一个栅栏，它有n个柱子，现在要给柱子染色，有k种颜色可以染。
必须保证不存在超过2个相邻的柱子颜色相同，求有多少种染色方案。

 注意事项

n和k都是非负整数

您在真实的面试中是否遇到过这个题？ 

Yes

样例

n = 3, k = 2, return 6

      post 1,   post 2, post 3
way1    0         0       1 
way2    0         1       0
way3    0         1       1
way4    1         0       0
way5    1         0       1
way6    1         1       0

这个问题，当染第3个柱子时，需要考虑第1个柱子与第2个柱子，如果1与2个柱子颜色一样，那么此时只有k*1*(k-1)种可能，如果不一样那就有k*(k-1)*k种可能，k*(k-1)*k=(k-1)*(k+k*(k-1))而k又是第一个柱子的染色可能数，k*(k-1)是第二个柱子的染色可能数，故有此可推倒出，状态转移方程为：dp[i]=(k-1)*(dp[i-1]+dp[i-2])

代码：

class Solution {
public:
    /*
     * @param n: non-negative integer, n posts
     * @param k: non-negative integer, k colors
     * @return: an integer, the total number of ways
     */
    int numWays(int n, int k) {
        // write your code here
        if(n==0)return 0;
        if(n==1)return k;
        if(n==2)return k*k;
        int a=k,b=k*k;
        for(int i=3;i<=n;i++)
        {
            int c=b;
            b=(a+b)*(k-1);
            a=c;
        }
        return b;
    }
};