lintcode——栅栏染色

我们有一个栅栏，它有n个柱子，现在要给柱子染色，有k种颜色可以染。
必须保证任意两个相邻的柱子颜色不同，求有多少种染色方案。
注意事项

n和k都是非负整数

样例

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

当有1个栅栏的时候，有k种；
当有2个栅栏的时候，有k*k种；
当有3个栅栏的时候，分两种情况：
a.如果3和2颜色相同，则有k*(k-1)。
b.如果3和2颜色不同，则有(k*k)*(k-1)。
则状态转移方程为：dp[i] = (k - 1) * (dp[i-1] + dp[i-2]);

package com.fk.question;

/**
 * Created by fengkai on 1/11/17.
 */
public class NumWays {

    public int numWays(int n, int k) {
        // f(i) = f(i-1) * (k-1) + f(i-2) * (k-1)
        int dp[] = {0, k , k*k, 0};
        if(n <= 2)
        {
            return dp[n];
        }
        for(int i = 2; i < n; i++)
        {
            dp[3] = (k - 1) * (dp[1] + dp[2]);
            dp[1] = dp[2];
            dp[2] = dp[3];
        }
        return dp[3];
    }

    public static void main(String[] args) {
        System.out.println(new NumWays().numWays(2,3));
    }

}