栅栏染色
我们有一个栅栏,它有n
个柱子,现在要给柱子染色,有k
种颜色可以染。
必须保证不存在超过2个相邻的柱子颜色相同,求有多少种染色方案。
注意事项
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
这个问题,当染第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;
}
};
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;
}
};