题目
有 k 种颜色的涂料和一个包含 n 个栅栏柱的栅栏,每个栅栏柱可以用其中一种颜色进行上色。
你需要给所有栅栏柱上色,并且保证其中相邻的栅栏柱 最多连续两个 颜色相同。然后,返回所有有效涂色的方案数。
注意:
n 和 k 均为非负的整数。
示例:
输入: n = 3,k = 2
输出: 6
解析: 用 c1 表示颜色 1,c2 表示颜色 2,所有可能的涂色方案有:
柱 1 柱 2 柱 3
----- ----- ----- -----
1 c1 c1 c2
2 c1 c2 c1
3 c1 c2 c2
4 c2 c1 c1
5 c2 c1 c2
6 c2 c2 c1
题解
Easy难度,我的思路是构建一个2k*n的二维数组(颜色x栅栏),每个颜色占两行,第一行储存上一个是其他颜色的栅栏,第一行储存上一个是同样颜色的栅栏,最后把所有可能相加即可
颜 | 1 | 1 | 2 |
色 | 0 | 1 | 1 |
| 1 | 1 | 2 |
| 0 | 1 | 1 |
| 栅 | 栏 | 数 |
class Solution {
public int numWays(int n, int k) {
if(n==0 || k==0) return 0;
if(n>2 && k==1) retu