Codeforces 528 D. Fuzzy Search (FFT)

链接：D. Fuzzy Search

题意：
给出一个母串S和一个模板串T，求模板串在母串中的匹配次数。
匹配时，如果用s[i]匹配t[j]，那么只要s[i-k]-s[i+k]中有字母与t[j]相同即可算作匹配成功。

思路：

1. 对于普通的字符串匹配，我们考虑每一个字符，只要把模板串倒过来，两个串中是该字符的位置设置为 1 ，跑一次 FFT , 求出每一个字符对模板穿移动 i 位的贡献，如果所有字母的贡献 如果等于模板串的长度，答案就+1.
2. 对于此题，如果母串中第 i 位为该字符，那么我们就把 [i - k , i + k]位都设为 1 。跑FFT 即可符合题意中的匹配。最后算贡献与普通的同理。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
const int num = 200002;
const double Pi = acos(-1.0);
int r[maxn];
char s[5] = {'A' , 'G' ,'C' , 'T'};
int T, n , m , k ,d[maxn] , vis[maxn];
char s1[maxn],s2[maxn];
int ans[maxn] , cnt[maxn];
struct Complex{
    double x, y;
    Complex(double xx = 0, double yy = 0) : x(xx), y(yy) {}
    Complex operator+(Complex b){
        return Complex(x + b.x, y + b.y);
    }
    Complex operator-(Complex b){
        return Complex(x - b.x, y - b.y);
    }
    Complex operator*(Complex b){
        return Complex(x * b.x - y * b.y, x * b.y + y * b.x);
    }
};

// 最高n，m次， 0次为常数
Complex a[maxn], b[maxn];
Complex c[maxn];
void fft(Complex *cm, int cnum, int tag){
    for(int i =  0 ; i <= cnum - 1; i ++){
        if (i < r[i]){
            swap(cm[i], cm[r[i]]);
        }
    }
    for (ll mid = 1; mid < cnum; mid <<= 1){
        Complex wk = Complex(cos(2 * Pi / (2 * mid)), tag * sin(2 * Pi / (2 * mid)));
        for (ll j = 0; j < cnum; j += 2 * mid) //枚举 cnum/2*mid个全长段
        {
            Complex w(1, 0);
            for (ll k = 0; k < mid; k++) //每段里面进行fft，不是<=，因为只有一半，不能超出
            {
                Complex buf = w * cm[j + k + mid];
                cm[j + k + mid] = cm[j + k] - buf; //在这一步cm[j + k] = cm[j + k] + buf上，否则cm[j+k]已被更改
                cm[j + k] = cm[j + k] + buf;
                w = w * wk;
            }
        }
    }
}

int fft_init(int n){     //n是最高次幂
    int maxx = 1, bits = 0;
    while ((maxx) <= n) {
        maxx <<= 1;
        bits++;
    }
    for(int i = 0; i <= maxx - 1; i ++){
        r[i] = (r[i >> 1] >> 1) | ((i & 1) << (bits - 1)); //求二进制反转结果
    }
    return maxx;
}
int nn;
void work(int maxx){
     fft(a , maxx , 1);
     fft(b , maxx , 1);
     for(int i = 0; i <= maxx; i ++){
        c[i] = a[i] * b[i];
     }
     fft(c , maxx , -1);
     for(int i = 0; i <= n - m; i ++){
        vis[i] = (c[i + n].x / maxx + 0.5);
       // printf ("%c %d %d\n",nn + 'a',i,vis[i]);
        cnt[i] += vis[i];
     }
     for(int i = 0; i <= maxx; i ++){
        a[i].x = b[i].x = c[i].x = 0;
        a[i].y = b[i].y = c[i].y = 0;
     }
}

int main(){
    scanf("%d%d%d",&n,&m,&k);
    scanf("%s%s", s1+ 1 , s2 + 1);
    int maxx = fft_init(m + 2 * n);
    for(int now = 0; now < 4; now ++){
        memset(d , 0 , sizeof(d));
        for(int i = 1; i <= n; i ++){
            if(s1[i] == s[now]){
                d[max(1 , i - k)]++;   //差分一下
                d[i + k + 1]--;
            }
        }
        for(int i = 1; i <= n; i ++) d[i] += d[i - 1];
        for(int i = 1; i <= n; i ++){
            if(d[i]) a[i].x = 1;
        }
        for(int i = 1; i <= m; i ++){
            if(s2[i] == s[now]){
                b[n - i].x = 1;
            }
        }
        work(maxx);
    }
    int ans = 0;
    for(int i = 0; i <= n - m; i ++){  //cnt[i]代表移动 i 位 后的匹配数
        if(cnt[i] == m) ans ++;
    }
    printf ("%d\n",ans);

    return 0;
}