栅栏染色—动态规划

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

 

posted @ 2017-09-27 10:28  豆炸儿  阅读(728)  评论(0编辑  收藏  举报