lintcode514-栅栏染色

题目描述：

刚看到题目也是不知道怎么做，就尝试了一下过程，发现有规律可循：

在第1根柱子，有K种方法；到了第2根柱子，因为允许可以有两个颜色重复，所以就有了K*K中方法；

但是到了第3根柱子，就要考虑分情况了：如果第3根柱子和第2根柱子的颜色相同，方法数就是第2根柱子的方法数，这时就要求第2根柱子和第1根不能是一样的颜色，所以方法数变成k*(K-1)种，这也是第3根柱子的方法数；如果第3根柱子和第2根柱子的颜色不相同，第2根柱子还是k*K种，第三根柱子要除过第2根柱子的颜色，于是就变成了k*K*(K-1)种；两种情况加起来就是(K+K*K)*(K-1);

可以看出这其实变成了动态规划的思想：buf[i]表示当柱子有i根时的方法数；

buf[1] = K;

buf[2] = k*k;

...

buf[i] = (buf[i-1] + buf[i-2]) * (K-1);

另一种解释：考虑buf[i]时，如果buf[i]和buf[i-1]颜色相同，则有buf[i-2] * (k-1);

                                        如果buf[i]和buf[i-1]颜色不相同，则有buf[i-1] * (k-1); 两种情况相加即可。

但是在具体代码过程中会发现，i位置的值只与i-1位置和i-2位置有关，用两个变量记录即可，不需要用数组把所有的情况都记录下来。

代码：

int numWays(int n, int k){
    if(n==1)
        return k;
    if(n==2)
        return k*k;

    int first = k;
    int second = k*k;
    int third = 0;
    for(int i=3; i<=n; ++i){
        third = (first + second) * (k-1);
        first = second;
        second = third;
    }
    return third;
}

参考博客：

https://blog.csdn.net/waltonhuang/article/details/52037127

https://blog.csdn.net/sinat_32197439/article/details/60352631