栅栏染色—动态规划

思路：第一根栅栏有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     }

 

转载于:https://www.cnblogs.com/changhongdou/p/7600404.html