题目:
思路:
第一想法是直接构建串,果不其然超时了。
先构造几个字符串看看规律,可以发现下一层的字符串=上一层字符串+上一层字符串的每一个取反,第i行的字符串长度为2**i(i从0开始)
进一步改进,已知串的构造规律,倒推第N行第K个字符是从哪一个转换过来的。
class Solution:
def kthGrammar(self, N: int, K: int) -> int:
s = [0]
# 记录取反的次数
times = 0
while N > 0 and K > 0:
N -= 1
# 上一层的长度小于K,说明K是上一层某字符取反后的结果
if 2**N < K:
K -= 2**N
times += 1
# 否则这个字符在第K行和第K-1行都是一样的
while K > 0:
s += [1^i for i in s]
K -= 1
return 1^s[N] if times % 2 else s[N]