残缺的字符串【FFT】

[luogu4173]残缺的字符串
求对于\(B\)的每一个位置 \(i\),从这个位置开始连续\(m\)个字符形成的子串是否可能与\(A\)串完全匹配.\(1\le m\le n \le 3*{10}^5\)
两个位置\(A_i\)\(B_{j+i}\)匹配有三种情况:

  1. \(A_i\)为'*'
  2. \(B_{i+j}\)为'*'
  3. \(A_i = B_{i+j}\),即\(A_i-B_{i+j}=0\)

那么显然可以令''所在位置为0,把三种情况乘起来就可以判断一个位置是否匹配了
判断从\(j\)开始的\(m\)个位置是否匹配就是$$S(j)=\sum_{i=1}m(A_i-B_{i+j})2
A_i*B_{i+j}$$

\[S(j)=\sum_{i=1}^m(A_i^2-2*A_i*B_{i+j}+B_{i+j}^2)*A_i*B_{i+j} \]

\[=\sum_{i=1}^mA_i^3*B_{i+j}-2\sum_{i=1}^mA_i^2*B_{i+j}^2+\sum_{i=1}^mA_i*B_{i+j}^3 \]

\(A\)翻转求卷积就好了

vector<int> q;
double eps=1e-6;
int dcmp(double x){if(fabs(x) < eps) return 0; return 1;}
char c[Maxn];
void solve(){
    m=read(), n=read(); scanf("%s", c+1);
    for(int i=1; i <= m; i++) f[1][m-i].a=c[i] == '*' ? 0 : c[i]-'a'+1,
    	f[2][m-i].a=f[1][m-i].a*f[1][m-i].a, f[3][m-i].a=f[1][m-i].a*f[2][m-i].a;
    scanf("%s", c);
    for(int i=0; i < n; i++) g[1][i].a=c[i] == '*' ? 0 : c[i]-'a'+1,
        g[2][i].a=g[1][i].a*g[1][i].a, g[3][i].a=g[1][i].a*g[2][i].a;
    while(lim <= n+m-2) lim<<=1, l++;
    for(int i=0; i < lim; i++) r[i]=(r[i>>1]>>1)|((i&1)<<l-1);
    for(int i=1; i <= 3; i++) FFT(f[i], 1), FFT(g[i], 1);
    for(int i=1; i <= 3; i++) for(int j=0; j < lim; j++) f[i][j]=f[i][j]*g[4-i][j];
    for(int i=1; i <= 3; i++) FFT(f[i], -1);
    double delta;
    for(int i=m-1; i < n; i++) {
        delta=(f[1][i].a-2*f[2][i].a+f[3][i].a)/lim;
        if(!dcmp(delta)) q.push_back(i-m+2); //printf("%d %lf\n", i, delta);
    }
    printf("%d\n", q.size());
    for(int i=0; i < q.size(); i++) printf("%d ", q[i]);
}

posted @ 2018-07-12 15:37  zerolt  阅读(220)  评论(0编辑  收藏  举报