[LeetCode] 779. 第K个语法符号

1 题目描述

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

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

2 解题思路

这道题目重在理解，可以先来计算表示上一行的长度，
计算出来之后，前半段和上一行位置的数相同，后半段和相同位置的数相反

3 解决代码

• java 代码

class Solution {
    public int kthGrammar(int N, int K) {
        if(N == 1){
            return 0;
        }
        
        //计算上一行的长度，也就是本行长度折半 
        int prevLen = (1 << (N - 1))/2;
        
        if (K<= prevLen){
            //位于前半段，则等于上一行相同位置的值
            return kthGrammar(N - 1, K);
        }
            
        else{
            //位于后半段，则等于上一行相同位置的值取反
            return 1 - kthGrammar(N - 1, K-prevLen );
        }
            
    }
}

• python代码

class Solution(object):
    def kthGrammar(self, N, K):
        """
        :type N: int
        :type K: int
        :rtype: int
        """
        #这道题目重在理解，前半段和上一行位置的数相同，后半段和相同位置的数相反
        if N == 1:
            return 0
        #计算上一行的长度，也就是本行长度折半 
        #" // " 表示整数除法,返回不大于结果的一个最大的整数
        prevLen = (1 << (N - 1)) // 2
        
        if K<= prevLen:
            #位于前半段，则等于上一行相同位置的值
            return self.kthGrammar(N - 1, K)
        else:
            #位于后半段，则等于上一行相同位置的值取反
            return 1-self.kthGrammar(N - 1, K-prevLen )