LeetCode题目笔记——779. 第K个语法符号，从超时到0ms（bushi）

这道题是今天的每日一题，还挺有意思的，所以记录一下

题目描述

我们构建了一个包含 n 行( 索引从 1 开始 )的表。首先在第一行我们写上一个 0。接下来的每一行，将前一行中的0替换为01，1替换为10。

例如，对于 n = 3 ，第 1 行是 0 ，第 2 行是 01 ，第3行是 0110 。 给定行数 n 和序数 k，返回第 n 行中第 k个字符。（ k 从索引 1 开始）

示例 1:

输入: n = 1, k = 1 输出: 0 解释: 第一行：0
示例 2:

输入: n = 2, k = 1 输出: 0 解释: 第一行: 0 第二行: 01
示例 3:

输入: n = 2, k = 2 输出: 1 解释: 第一行: 0 第二行: 01

来源：力扣（LeetCode） 链接：https://leetcode.cn/problems/k-th-symbol-in-grammar
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目难度——中等

方法一：模拟

  根据题意，我们可以模拟这个变化的过程，n从1开始，具体的，我们可以用一个循环，从1到n，对每个数也即每一行按照规则生成一个新的字符串，最后得到第n行字符串，然后我们再返回这第n行字符串的第k个数就好。

代码/Python

class Solution:
    def kthGrammar(self, n: int, k: int) -> int:
        res = "0"
        table = {'0':'01', '1':'10'}
        for i in range(1, n + 1):
            tmp = ""
            for w in res:
                tmp += table[w]
            res = tmp
        
        return int(res[k - 1])

  注意题目中的提示，每行的字符数是指数级别提升的，所以上面这个暴力模拟方法肯定会超时。而且第n+1行有一半的无效操作，具体原因见后文分析。

方法二：观察规律

  我们可以手动模拟一下这个变化过程：

n = 1: 0
n = 2: 01
n = 3: 0110
n = 4: 01101001
n = 5: 0110100110010110
…

  可以发现，第n行一共有2的n-1次方个数，而且第n+1行的前半部分和第n行一模一样，同时后半部分就是前半部分按位取反，那么我们可以再试试用这个规律来加速上面那个暴力过程？很遗憾，同样会超时。
  （下标从1开始）所以我们需要另辟蹊径，以第5行为例，0110100110010110。假如k=16，也就是最后一个数，是0，如果k=5，就是倒数第二个数，即1，根据说的规律，我们可不可以用类似二分法的思路来做？k=16，最后一个数由第八位数反转而来，第八个数又是由第四位数反转而来，第四位数又是由第二位数反转，第二位由由第一位反转而来。如果我们用二分来表示这个过程：

16 / 2 = 8
8 / 2 = 4
4 / 2 = 2
2 / 2 = 1
1 / 1 = 0

  一直到0，我们经历了5次除法，或者说5次反转。再看看k=15的时候：（默认整数除法）

15 / 2 = 7
7 / 2 = 3
3 / 2 = 1
1 / 2 = 0

  直到0，我们经历了4次除法，或者说4次反转。

代码/Python

  代码借助了一下官方题解，思路是一样的，自己写的就是一坨屎，还有bug，这里就不放了

class Solution:
    def kthGrammar(self, n: int, k: int) -> int:
        k -= 1
        res = 0
        while k:
            k &= k - 1
            res ^= 1
        return res

总结

  第一种方法无疑是O(N2)的，第二种方法借用了二分的思想，所以是O(logN)。