[CF1056E]Check Transcription

题目：Check Transcription

传送门：http://codeforces.com/contest/1056/problem/E

分析：

1）显然有个$O( \frac{t}{max(cnt_0,cnt_1)}*t)$暴力：枚举替代$0$的字符串，计算替代$1$的字符串长度，判断是否可行。

2）我们字符串匹配不必一个一个字符比较，可以用哈希。这样复杂度就降为了$O( \frac{t}{max(cnt_0,cnt_1)}*s)$,由于$max(cnt_0,cnt_1)>=\frac{s}{2}$,这样复杂度就大约是$O(t)$

3）哈希基于概率，有可能会冲突，简单的做法是双哈希，就是这么暴力，如果还不行，那就再来一遍。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ULL;
const int maxN=1e6+10;
const ULL HAP[2]={567,568};
ULL ts[2][2],fac[2][maxN],Ha[2][maxN];
char s[maxN],t[maxN];
int ans,cnt[2];
bool check(int op,int l,int r){
    bool pd=true;
    for(int j=0;j<2;++j)
        pd&=ts[op][j]==Ha[j][r]-Ha[j][l-1]*fac[j][r-l+1];
    return pd;
}
void HashInit(int n){
    for(int j=0;j<2;++j){
        fac[j][0]=1;for(int i=1;i<=n;++i)fac[j][i]=fac[j][i-1]*HAP[j];
        for(int i=1;i<=n;++i)Ha[j][i]=Ha[j][i-1]*HAP[j]+t[i];
    }
}
int main(){
    scanf("%s%s",s+1,t+1);
    int ls=strlen(s+1),lt=strlen(t+1);
    for (int i=1;i<=ls;++i)++cnt[s[i]-'0'];
    HashInit(lt);
    for (int l0=1,l1;l0*cnt[0]<lt;++l0)if ((lt-l0*cnt[0])%cnt[1]==0){
        l1=(lt-l0*cnt[0])/cnt[1];
        bool p0=1,p1=1,p=1;int k=1;
        for(int i=1;i<=ls;++i){
            if (!p) break;
            if (s[i]=='0'){
                if(p0){
                    p0=0;
                    for(int j=0;j<2;++j)ts[0][j]=Ha[j][k+l0-1]-Ha[j][k-1]*fac[j][l0];
                    if(!p1 && l0==l1) p=!check(1,k,k+l0-1);
                }else p=check(0,k,k+l0-1);
                k+=l0;
            }else{
                if(p1){
                    p1=0;
                    for(int j=0;j<2;++j)ts[1][j]=Ha[j][k+l1-1]-Ha[j][k-1]*fac[j][l1];
                    if(!p0 && l0==l1) p=!check(0,k,k+l1-1);
                }else p=check(1,k,k+l1-1);
                k+=l1;
            }
        }
        if (p) ++ans;
    }
    printf("%d",ans);
    return 0;
}