递归 | 第k个语法符号(Python)

最新推荐文章于 2022-10-22 13:06:16 发布

原创最新推荐文章于 2022-10-22 13:06:16 发布 · 478 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#递归算法 #leetcode

该博客介绍了如何解决LeetCode上的一个问题——找出给定行数N和序数K的第N行第K个字符。通过递归算法，利用位异或的特性，可以巧妙地解决这个问题。当K为奇数时，对应'1'，偶数时对应'0'。博客还提供了问题的链接和解决方案参考。

在第一行我们写上一个 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)].

链接：https://leetcode-cn.com/problems/k-th-symbol-in-grammar

暴力法

class Solution(object):
    def kthGrammar(self, N, K):
        rows = []
        lastrow = '0'
        while len(rows) < N:
            rows.append(lastrow)
            lastrow = "".join('01' if x == '0' else '10'
                              for x in lastrow)
        return int(rows[-1][K-1])

递归法(父子递归）

class Solution(object):
    def kthGrammar(self, N, K):
        if N == 1: return 0
        return (1 - K%2) ^ self.kthGrammar(N-1, (K+1)//2)
        #第 K 个数字是上一行第 (K+1) // 2 个数字生成的, (1-K%2)是因为如果上一行的数字为 0，被生成的数字为 1 - (K%2)，如果上一行的数字为 1，被生成的数字为 K%2.

位异或用的很巧妙，因为0，1生成的’0’,‘1’顺序刚好相反，因此只要一个判断条件，相对应的，位异或看K%2，得到的数字刚好就是奇数位’1’,偶数位’0’, 如果上一行的相应数字是1，则得到奇数位’0’,偶数位’1’，即对应’01’,矛盾，故选择 1-K%2.

转换为二进制运算

class Solution(object):
    def kthGrammar(self, N, K):
        return bin(K - 1).count('1') % 2  #翻转的次数等于 K-1 二进制表示形式中 1 出现的个数

参考：https://leetcode-cn.com/problems/k-th-symbol-in-grammar/solution/di-kge-yu-fa-fu-hao-by-leetcode/