514. 栅栏染色

514. 栅栏染色

 

 

我们有一个栅栏，它有 n 个柱子，现在要给柱子染色，有 k 种颜色可以染。

必须保证不存在超过2个相邻的柱子颜色相同，求有多少种染色方案。

样例

例 1:

输入: n=3, k=2

输出: 6

Explanation:

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

例 2:

输入: n=2, k=2

输出: 4

Explanation:

post 1, post 2

way1 0 0

way2 0 1

way3 1 0

way4 1 1

注意事项

n 和 k 都是非负整数

public class Solution {

    /**

     * @param n: non-negative integer, n posts

     * @param k: non-negative integer, k colors

     * @return: an integer, the total number of ways

     */

    public int numWays(int n, int k) {

        // write your code here

          if (n<1)return 0;

            if (n==1)return k;

            int[] ret=new int[n];

            ret[0]=k;

            ret[1]=k*k;

            for (int i = 2; i < n; i++) {

                ret[i] = ret[i-1] * (k-1) + ret[i-2] * (k-1);

            }

            return ret[n-1];

    }

}