Leetcode 779. 第K个语法符号 / python实现

题目:

思路:

第一想法是直接构建串,果不其然超时了。

先构造几个字符串看看规律,可以发现下一层的字符串=上一层字符串+上一层字符串的每一个取反,第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]

Guess you like

Origin blog.csdn.net/qq_42711381/article/details/105564930