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;
  }
};