[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+;
const ULL HAP[]={,};
ULL ts[][],fac[][maxN],Ha[][maxN];
char s[maxN],t[maxN];
int ans,cnt[];
bool check(int op,int l,int r){
    bool pd=true;
    for(int j=;j<;++j)
        pd&=ts[op][j]==Ha[j][r]-Ha[j][l-]*fac[j][r-l+];
    return pd;
}
void HashInit(int n){
    for(int j=;j<;++j){
        fac[j][]=;for(int i=;i<=n;++i)fac[j][i]=fac[j][i-]*HAP[j];
        for(int i=;i<=n;++i)Ha[j][i]=Ha[j][i-]*HAP[j]+t[i];
    }
}
int main(){
    scanf("%s%s",s+,t+);
    int ls=strlen(s+),lt=strlen(t+);
    for (int i=;i<=ls;++i)++cnt[s[i]-''];
    HashInit(lt);
    for (int l0=,l1;l0*cnt[]<lt;++l0)if ((lt-l0*cnt[])%cnt[]==){
        l1=(lt-l0*cnt[])/cnt[];
        bool p0=,p1=,p=;int k=;
        for(int i=;i<=ls;++i){
            if (!p) break;
            if (s[i]==''){
                if(p0){
                    p0=;
                    for(int j=;j<;++j)ts[][j]=Ha[j][k+l0-]-Ha[j][k-]*fac[j][l0];
                    if(!p1 && l0==l1) p=!check(,k,k+l0-);
                }else p=check(,k,k+l0-);
                k+=l0;
            }else{
                if(p1){
                    p1=;
                    for(int j=;j<;++j)ts[][j]=Ha[j][k+l1-]-Ha[j][k-]*fac[j][l1];
                    if(!p0 && l0==l1) p=!check(,k,k+l1-);
                }else p=check(,k,k+l1-);
                k+=l1;
            }
        }
        if (p) ++ans;
    }
    printf("%d",ans);
    return ;
}