在第一行我们写上一个 0。接下来的每一行,将前一行中的0替换为01,1替换为10。
给定行数 N 和序数 K,返回第 N 行中第 K个字符。(K从1开始)
例子:
输入: N = 1, K = 1
输出: 0
输入: N = 2, K = 1
输出: 0
输入: N = 2, K = 2
输出: 1
输入: N = 4, K = 5
输出: 1
解释:
第一行: 0
第二行: 01
第三行: 0110
第四行: 01101001
注意:
- N 的范围 [1, 30].
- K 的范围 [1, 2^(N-1)]
思路一:
观察规律可得,N层的第K个数是由第N-1层的第(K+1)/2个数转化而来,设N-1层的第
(
K
+
1
)
/
2
(K+1)/2
(K+1)/2个数值为
v
1
v_1
v1,N层的第K个数值为
v
2
v_2
v2,则满足:
v
2
=
1
−
(
K
%
2
)
⊗
v
1
v_2=1-(K\%2) \otimes v_1
v2=1−(K%2)⊗v1, 根绝两者之间的关系可以写出递归代码:
class Solution {
//定义kthGramma含义为N层第K个数的值
public int kthGrammar(int N, int K) {
if (N == 1) return 0; //终止条件
return 1-((K%2) ^ kthGrammar(N-1, (K+1)/2));
}
}
思路二:
每一层的数都是由上一层得到,可以迭代将每层数求到:
class Solution {
public int kthGrammar(int N, int K) {
int[] lastrow = new int[1 << N];
for (int i = 1; i < N; ++i) {
for (int j = (1 << (i-1)) - 1; j >= 0; --j) { //迭代时,为了防止当前层的左面覆盖掉右面的数,需要从右往左以此赋值
lastrow[2*j] = lastrow[j];
lastrow[2*j+1] = 1 - lastrow[j];
}
}
return lastrow[K-1];
}
}
这种方式时间复杂度和空间复杂度都达到 O ( 2 N ) O(2^N) O(2N),会造成内存溢出,不推荐,也无法通过较大N和K的测试用例。