URAL 1996 Cipher Message 3

题目

神题。

记得当初DYF和HZA讲过一个FFT+KMP的题目，一直觉得很神，从来没去做。

没有真正理解FFT的卷积。

首先考虑暴力。

只考虑前7位 KMP 找出所有 B 串可以匹配 A 串的位置。

设 a(i) = A(i) & 1, b(i) = B(i) & 1

然后相当于求所有的

c(i) =  ∑_k=0^m-1 a(i+k) * b(k)

考虑卷积形式:

c(i) =  ∑_k=0^m-1 a(k) * b(i - k)

将b串反过来

c(i) = ∑_k=0^m-1  a(k) * b(m-i+k)

改变 c 的定义令原先的 C(i) = 现在的 c(i+m)

得到 C(i) = ∑_k=0^m-1  a(k) * b(k-i)

所以 C(i) = ∑_k=0^m  a(k) * b(i-k+n)

然后FFT解决。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <complex>
#include <cmath>

#define N 250010
#define Ex complex<double>
#define pi 3.14159265354

using namespace std;

int n,m,a[N],b[N],same[N],rev[N<<];

inline void scan(int &x){
    char tmp[];
    scanf("%s",tmp);
    x=;
    int len=strlen(tmp);
    for(int i=len-;~i;i--)
        if(tmp[i]=='') x|=(<<(len-i-));
}

inline bool simple(int a,int b){
    return (a|)==(b|);
}

int f[N];
bool match[N];
Ex A[N<<],B[N<<],C[N<<];

void fft(Ex x[],int n,int t){
    for(int i=;i<n;i++){
        if(rev[i]>i) swap(x[rev[i]],x[i]);
    }
    for(int m=;m<n;m<<=){
        Ex wn(cos(pi/m*t),sin(pi/m*t));
        for(int k=;k<n;k+=(m<<)){
            Ex wt(,);
            for(int i=;i<m;i++,wt*=wn){
                Ex &A=x[i+m+k],&B=x[i+k],tmp=wt*A;
                A=B-tmp; B=B+tmp;
            }
        }
    }
    if(t==-){
        for(int i=;i<n;i++)
            x[i]/=(double)n;
    }
}

int main(){
    freopen("message10.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=;i<n;i++) scan(a[i]);
    for(int i=;i<m;i++) scan(b[i]);
    f[]=f[]=;
    for(int i=;i<m;i++){
        int j=f[i];
        while(j&&!simple(b[i],b[j])) j=f[j];
        f[i+]= simple(b[i],b[j])? j+:;
    }
    int j=;
    bool flag=;
    for(int i=;i<n;i++){
        while(j&&!simple(b[j],a[i])) j=f[j];
        if(simple(b[j],a[i])) j++;
        if(j==m) match[i-m+]=,flag=;
    }
    if(!flag){
        puts("No");
        return ;
    }
    puts("Yes");
    int T=,nt;
    for(nt=;nt<=(n+m);nt<<=) T++;
    for(int i=;i<nt;i++) rev[i]=(rev[i>>]>>)|((i&)<<(T-));
    for(int i=;i<n;i++) A[i]=a[i]&;
    for(int i=;i<m;i++) B[i]=b[m-i-]&;
    fft(A,nt,);
    fft(B,nt,);
    for(int i=;i<nt;i++) C[i]=A[i]*B[i];
    fft(C,nt,-);
    for(int i=;i<n;i++) same[i]+=(int)(C[i].real()+0.5);
    for(int i=;i<nt;i++) A[i]=B[i]=C[i]=;
    for(int i=;i<n;i++) A[i]=(a[i]&)^;
    for(int i=;i<m;i++) B[i]=(b[m-i-]&)^;
    fft(A,nt,);
    fft(B,nt,);
    for(int i=;i<nt;i++) C[i]=A[i]*B[i];
    fft(C,nt,-);
    for(int i=;i<n;i++) same[i]+=(int)(C[i].real()+0.5);
    int ansv=0x3f3f3f3f,anst=;
    for(int i=;i<n;i++){
        if(!match[i]) continue;
        if(m-same[i+m-]<ansv){
            ansv=m-same[i+m-];
            anst=i+;
        }
    }
    printf("%d %d\n",ansv,anst);
    return ;
}