题目
在第一行我们写上一个 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)].
解析
遇到这种题目第一感觉就是要找到规律,寻找循环体。但是还是很抽象的,所以先尝试了暴力法也就是用ArrayList动态加入元素,方法可行,但是内存消耗巨大,不可取不可取~,还是安心找规律吧。
假设我们求第4行的第4个数是0还是1,那么第4行的第4个数字 m 就是根据第3行的第2个数 n 而来,若n=0, 那么 m = 1,否则m = 0;n 又是按照第 2 行的第1个数而来,而第一行是设定好的为0。我们可以推广到更大的数字比如第10行的第1000个数字,就可以按此规律来推出。最终我们写出了循环方法体:
当K为偶数时,第K个数就为 N-1行的 K / 2 == 0?1:0;
当K为奇数时,第K个数就为N-1行的 (K+1)/ 2 == 0?0:1;
Java代码
class Solution {
public int kthGrammar(int N, int K) {
if(N == 1) return 0;
if(K % 2 == 0) return kthGrammar(N-1, K / 2) == 0 ? 1:0;
else return kthGrammar(N-1,(K+1) / 2) == 0 ? 0:1;
}
}