字符串哈希——1056E

典型的字符串哈希题

字符串hash[i]：子串s[1-i]代表的值

要截取串s[l-r]  求hash[r]-hash[l-1]*F[len]即可

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define P 131
#define mod 1000000007
#define maxn 3000005
char s[maxn],t[maxn];
ll cnt0,cnt1,len1,len2,n,F[maxn],ash[maxn];

int check(int a,int b){
    int last=;
    ll h1,f1=,h2,f2=;
    for(int i=;i<=len1;i++){
        if(s[i]==''){
            if(f1==)h1=(ash[last+a]-ash[last]*F[a]%mod+mod)%mod,f1=;
            else if(h1!=(ash[last+a]-ash[last]*F[a]%mod+mod)%mod) return ;
            last+=a;
        }
        else {
            if(f2==)h2=(ash[last+b]-ash[last]*F[b]%mod+mod)%mod,f2=;
            else if(h2!=(ash[last+b]-ash[last]*F[b]%mod+mod)%mod) return ;
            last+=b;
        }
    }
    if((h1+mod)%mod==(h2+mod)%mod)return ;
    return ;
}

int main(){
    F[]=;
    for(int i=;i<=;i++)F[i]=F[i-]*P%mod;
    cin>>s+>>t+;
    len1=strlen(s+);len2=strlen(t+);
    for(int i=;i<=len1;i++)
        if(s[i]-'')cnt1++;else cnt0++;
    for(int i=;i<=len2;i++)
        ash[i]=t[i]-'a'++ash[i-]*P,ash[i]%=mod;

    int ans=;
    for(int i=;i<=len2;i++){
        int t0=i,t1;
        if(t0*cnt0>=len2)break;
        if((len2-t0*cnt0)%cnt1==){
            t1=(len2-t0*cnt0)/cnt1;
            if(check(t0,t1))ans++;
        }
    }
    cout<<ans<<endl;
}