
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 pre=k;
int now=k*k;
for(int i=3;i<=n;i++) {
int temp=now;
now=(pre+now)*(k-1);
pre=temp;
}
return now;
}
};
浙公网安备 33010602011771号