1545.找出第N个二进制字符串中的第k位
给你两个正整数 n 和 k,二进制字符串 Sn 的形成规则如下:
S1 = “0”
当 i > 1 时,Si = Si-1 + “1” + reverse(invert(Si-1))
其中 + 表示串联操作,reverse(x) 返回反转 x 后得到的字符串,而 invert(x) 则会翻转 x 中的每一位(0 变为 1,而 1 变为 0)。
例如,符合上述描述的序列的前 4 个字符串依次是:
S1 = “0”
S2 = “011”
S3 = “0111001”
S4 = “011100110110001”
请你返回 Sn 的 第 k 位字符 ,题目数据保证 k 一定在 Sn 长度范围以内。
示例 1:
输入:n = 3, k = 1
输出:“0”
解释:S3 为 “0111001”,其第 1 位为 “0” 。
示例 2:
输入:n = 4, k = 11
输出:“1”
解释:S4 为 “011100110110001”,其第 11 位为 “1” 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/find-kth-bit-in-nth-binary-string
import numpy as np
class Solution(object):
def findKthBit(self, n, k):
"""
:type n: int
:type k: int
:rtype: str
"""
result = []
if n == 1:
result.append(0)
for i in range(n-1):
if i == 0:
result.append(0)
temp = 1 - np.array(result[::-1])
temp = temp.tolist()
result.append(1)
result += temp
return str(result[k-1])