11/5/2018模拟 Problem C

Posted 2021-01-17 ubospica

题面

题解

我有特殊的哈希技巧

以到下一个相同字符的距离为值哈希, 如果不存在或在串外, 就是 (|T| + 1).

加入一个新字符 (S_i) 时, 同时修改它上一次出现时的值, 由 (|T| + 1) 修改为 (i).

详见代码.

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
#define rep(i,l,r) for(register int i=(l);i<=(r);++i)
#define repdo(i,l,r) for(register int i=(l);i>=(r);--i)
#define il inline
typedef double db;
typedef long long ll;
typedef unsigned long long ull;

//---------------------------------------
const int nsz=1e6+50,msz=1e5+50,base=1e5+3;
int n,m;
char s[nsz],t[msz];
il int tr(char c){return c-'a'+1;}
int nes[nsz],prs[nsz];
int net[msz],prt[msz];
ull hs=0,ht=0,pb[msz]{1};

void gene(char *s,int *ne,int *pr,int n){
    static int bot[30];
    rep(i,1,26)bot[i]=n+1;
    repdo(i,n,1){
        ne[i]=bot[tr(s[i-1])];
        bot[tr(s[i-1])]=i;
    }
    rep(i,1,26)bot[i]=0;
    rep(i,1,n){
        pr[i]=bot[tr(s[i-1])];
        bot[tr(s[i-1])]=i;
    }
}

void sol(){
    gene(s,nes,prs,n);
    gene(t,net,prt,m);
    rep(i,1,m){
        ht=ht*base+(net[i]==m+1?m+1:net[i]-i);
        hs=hs*base+(nes[i]>m?m+1:nes[i]-i);
        pb[i]=pb[i-1]*base;
    }
    if(hs==ht)cout<<1<<'
';
    ull tmp;
    rep(i,m+1,n){//[i-m+1,i]
        tmp=(nes[i-m]>i-1?m+1:nes[i-m]-(i-m));
        hs=hs*base-tmp*pb[m];
        tmp=i-prs[i];
        if(tmp<m){
            hs-=(m+1-tmp)*pb[tmp];   //dynamically modify
        }
        hs+=m+1;
        if(hs==ht)cout<<i-m+1<<'
';
    }
}

int main(){
//  freopen("c.in","r",stdin);
//  freopen("c.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0);
    cin>>s>>t;
    n=strlen(s),m=strlen(t);
    if(n<m)return 0;
    sol();
    return 0;
}