- 找出第 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” 。
示例 3:
输入:n = 1, k = 1
输出:“0”
示例 4:
输入:n = 2, k = 3
输出:“1”
提示:
1 <= n <= 20
1 <= k <= 2n - 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-kth-bit-in-nth-binary-string
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我们观察可以发现
- 对于
n<=4
时,我们可以直接查表
返回 - 对于
n>4
时,我们可以转化成n-1
的返回(设返回值ret)
,注意把返回值ret
和1
亦或(即ret^=1)
-
- 对于
k小于n/2
的情况,我们转化为求(n-1,k)
的情况
- 对于
-
- 对于
k大于n/2
的情况,我们转化成求(n-1, n-k)
,因为该字符串一定时对称取反
的(注意下标转化)
- 对于
-
-
01
1
100110110
0
01
01\color{red}{1} \color{black}100110110\color{red}{0}\color{black}01
011100110110001其中标红的是对称的,且反转的,所以下标变化就就是
n-k
代码如下
-
01
1
100110110
0
01
01\color{red}{1} \color{black}100110110\color{red}{0}\color{black}01
011100110110001其中标红的是对称的,且反转的,所以下标变化就就是
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif
#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#define MAXN ((int)1e5+7)
#define ll long long int
#define QAQ (0)
using namespace std;
#define num(x) (x-'0')
#define show(x...) \
do { \
cout << "\033[31;1m " << #x << " -> "; \
err(x); \
} while (0)
void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }
#define N ((1<<n))
class Solution {
public:
string sn[8] = {
"",
"00",
"0011",
"00111001",
"0011100110110001"
};
char findKthBit(int n, int k) {
if(n <= 4) return sn[n][k];
if(n == 1 || k == N-1 || k == (1<<(n-1))) return '1';
if(k <= (N/2)) return findKthBit(n-1, k);
else return (1^(findKthBit((n-1), N-k) ) );
}
};
#ifdef debug
signed main() {
freopen("test", "r", stdin);
clock_t stime = clock();
Solution s;
cout << s.findKthBit(5, 16) << endl;
clock_t etime = clock();
printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
return 0;
}
#endif