我们构建了一个包含 n 行( 索引从 1 开始 )的表。首先在第一行我们写上一个 0。接下来的每一行,将前一行中的0替换为01,1替换为10。例如,对于 n = 3 ,第 1 行是 0 ,第 2 行是 01 ,第3行是 0110 。给定行数 n 和序数 k,返回第 n 行中第 k 个字符。( k 从索引 1 开始)
思路
从上图可以看出,下层中的值是由上层的0
或者1
生成的。具体在下层中的值是0
还是1
由两个因素决定:
- 上层是
0
还是1
- 下层的位置的奇偶性
- 上层是
0
- 位置为奇数:下层值为: 0
- 位置为偶数:下层值为: 1
- 上层是
1
- 位置为奇数:下层值为: 1
- 位置为偶数:下层值为: 0
- 上层是
所以求n
行中第k
个字符可以转化成n-1
行第(k + 1) / 2
个位置的字符,转化成n - 2
行第(((k + 1) / 2) + 1) / 2
个位置的字符。。。。。。最终转成第1行第1个字符,而第一行第一个字符0
是已知条件,所以可以倒推出第n
行第k
个字符。这是明显的递归思路。
代码
function kthGrammar(n: number, k: number): number {
if (n === 1) return 0
const last = kthGrammar(n - 1, (k + 1) >> 1)
return last === 0 ? (k % 2 === 0 ? 1 : 0) : k % 2 === 0 ? 0 : 1
}
这里可以做些编程技巧上的简化,但是为了可读性就维持这样。