栅栏染色—动态规划
思路:第一根栅栏有k种染色方案,第二根也有k种,前两根染色方案有k*k种。第三根栅栏染色有k-1种,若第三根栅栏染色和第二根栅栏不同,那么第三个染色和第一根染色没有关系,则前三根栅栏染色方案有k*k*(k-1)种;若第三根栅栏和第二根栅栏染色相同,那么要求第一个栅栏和第三根栅栏染色不同,则前三根栅栏染色方案有k*(k-1)。
递推:现在考虑前i根栅栏染色方案。
若第i根栅栏和第i-1根栅栏颜色不同,则第i根栅栏和第i-2根栅栏染色无关,设前i-1根栅栏染色方案为m[i-1],此时第i根栅栏有k-1种染色可染,所以前i根染色方案有m[i-1]*(k-1)种。;
若第i根栅栏和第i-1根栅栏颜色相同,此时要求第i、i-1根栅栏和i-2根栅栏颜色不同,设前i-2根栅栏染色方案为m[i-2],此时第i根栅栏和第i-1根栅栏有k-1种染色可染,所以前i根染色方案有m[i-2]*(k-1)种。
1 int numWays(int n, int k) { 2 if (n == 0) { 3 return 0; 4 } 5 if (n == 1) { 6 return k; 7 } 8 if (n == 2) { 9 return k*k; 10 } else { 11 int * m = new int[n]; 12 m[0] = k; 13 m[1] = k*k; 14 for (int i = 2; i < n; i++) { 15 m[i] = m[i-1] * (k-1) + m[i-2] * (k-1); 16 } 17 return m[n-1]; 18 } 19 }