1. 问题描述:
给定一个只包含一个数字 1 的序列 [1]。现在,要对该序列进行 n−1 次扩充。每次扩充时,首先要将序列本身添加到其自身的尾部,然后还要在两个序列中间插入还未使用过的最小正整数。例如,序列 [1] 经过一次扩充后,得到序列 [1,2,1],再经过一次扩充后,得到序列 [1,2,1,3,1,2,1]。现在,请你计算在经过 n − 1 次扩充后,序列的第 k 个元素的值是多少?(元素从 1 开始编号)
输入格式
共一行,包含两个整数 n 和 k。
输出格式
输出一个整数,表示经过 n−1 次扩充后,序列的第 k 个元素的值。
数据范围
对于前三个测试点,1 ≤ n,k ≤ 10。
对于全部测试点,1 ≤ n ≤ 50,1 ≤ k ≤ 2 ^ n−1。
输入样例1:
3 2
输出样例1:
2
输入样例2:
4 8
输出样例2:
4
样例解释
对于样例 1,经过 2 次扩充,得到序列 [1,2,1,3,1,2,1],其第 2 个元素为 2。对于样例 2,经过 3 次扩充,得到序列 [1,2,1,3,1,2,1,4,1,2,1,3,1,2,1],其第 8 个元素为 4。
来源:https://www.acwing.com/problem/content/3698/
2. 思路分析:
我们可以找一下规律,可以发现第n个序列的长度为2 ^ n - 1,中间位置的元素为2 ^ (n - 1),所以当k = 2 ^ (n - 1)直接返回n即可,当k < 2 ^ (n - 1)时第k个位置在第n - 1个序列中,当k > 2 ^ (n - 1)时由于左右两边是对称的所以在第n - 1个序列的对称位置k - 2 ^ (n - 1),可以发现其实是一个递归的过程,直接分情况递归即可。
3. 代码如下:
class Solution:
def dfs(self, n: int, k: int):
if k == 1 << n - 1:
return n
elif k < 1 << n - 1:
return self.dfs(n - 1, k)
else:
return self.dfs(n - 1, k - (1 << n - 1))
def process(self):
n, k = map(int, input().split())
return self.dfs(n, k)
if __name__ == '__main__':
print(Solution().process())