链接:E. A Bit Similar
题意:
给一个 01 串 S,你需要找到一个字典序最小的长度为 k 的 01 串 T,使 T 与 S 中每一个子串相似(相似的定义是 两个串至少有一个位置相同)。
思路:
首先我们考虑不满足串有哪些 ,其实就是把 S 串取反,这时 S串中的每一个长度为 k 的子串都是不满足的,这些串最多只有 n - k 个,而可能的情况一共是 2 ^ k 个,把这个 01 串看成一个数的二进制 ,其实我们只要从 0 开始枚举(要求字典序最小) ,枚举到 n - k + 1 必然能找到一个合法的(前提是 n - k + 1 <= 2 ^ k),所以我们只要用个 vis 标记 n - k 个子串转换为10进制的值(超过 1e6的可以不用管),然后从 0 开始枚举找到第一个没有出现过的值就好了,如果到 2 ^ k - 1 还没找到 就是 NO。子串转 10 进制可以用 hash 维护。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 7;
const int rr=2;
const int mod = 998244353;
int T;
ll pre[maxn],poww[maxn],num[maxn];
char str[maxn];
int a[maxn];
int vis[maxn];
int n,k;
ll hashp(int l ,int r){
return (pre[r] - pre[l - 1] * poww[r - l + 1] % mod + mod) % mod;
}
int main(){
scanf("%d",&T);
poww[0] = 1;
for(int i = 1;i <= 30;i++) poww[i] = poww[i - 1] * rr % mod;
while(T--){
scanf("%d%d",&n,&k);
scanf("%s",str+1);
memset(vis,0,sizeof(vis));
pre[0]= num[0] = 0;
for(int i = 1;i <= n; i++){
a[i] = ((str[i] - '0') ^ 1);
pre[i]=(pre[i-1] * 2 + a[i])% mod;
num[i] = num[i - 1] + a[i];
}
for(int i = 1;i + k - 1 <= n; i ++ ){
int x = hashp(max(i , i + k - 20),i + k - 1);
if(k > 20 && num[i + k - 20] - num[i - 1] > 0) continue;// 超过了 1e6 ,20位以上 有 1
if(x > n) continue; //超过 n 没必要记录
vis[x] = 1;
}
int x = -1;
string ans = "";
for(int i = 0; i <= (1 << min(k , 21)) - 1; i ++){
if(vis[i] == 0){
x = i;
break;
}
}
if(x == -1) printf ("NO\n");
else{
printf ("YES\n");
for(int i = 0;i <= k; i ++) ans += '0';
for(int i = 20;i >= 0;i --){
if((x >> i) & 1) ans += '1';
else ans += '0';
}
cout<<ans.substr(ans.size() - k,ans.size())<<endl;
}
}
return 0;
}