Educational Codeforces Round 101 (Rated for Div. 2) - E. A Bit Similar(哈希 + 暴力)

链接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;
}

猜你喜欢

转载自blog.csdn.net/hddddh/article/details/111936833
今日推荐