CF528D Fuzzy Search

题意：给定k，只含有ACGT的字符串S和T，求T在S中出现了多少次。

字符匹配：如果S的[i - k, i + k]中有字符x，那么第i位可以匹配x。

解：

首先预处理：f[i][j]表示S的第i位能否匹配j。差分一下即可。

然后按照FFT的套路，枚举每种字符，算一遍有多少个匹配。四种字符加起来，如果匹配数等于T的长度，就匹配成功。

#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdio>
const int N = 200010;
const double pi = 3.1415926535897932384626;
const char ch[] = {'A', 'C', 'G', 'T'};
struct cp
{
    double x, y;
    cp(double X = 0, double Y = 0)
    {
        x = X;
        y = Y;
    } inline cp operator +(const cp &w) const
    {
        return cp(x + w.x, y + w.y);
    } inline cp operator -(const cp &w) const
    {
        return cp(x - w.x, y - w.y);
    } inline cp operator *(const cp &w) const
    {
        return cp(x * w.x - y * w.y, x * w.y + y * w.x);
    }
} a[N << 2], b[N << 2];
int r[N << 2], ans[N << 2], f[N][4], d[N];
char s[N], str[N];
inline void FFT(int n, cp *a, int f)
{
    for(int i = 0; i < n; i++)
    {
        if(i < r[i])
        {
            std::swap(a[i], a[r[i]]);
        }
    }
    for(int len = 1; len < n; len <<= 1)
    {
        cp Wn(cos(pi / len), f * sin(pi / len));
        for(int i = 0; i < n; i += (len << 1))
        {
            cp w(1, 0);
            for(int j = 0; j < len; j++)
            {
                cp t = a[i + len + j] * w;
                a[i + len + j] = a[i + j] - t;
                a[i + j] = a[i + j] + t;
                w = w * Wn;
            }
        }
    }
    if(f == -1)
    {
        for(int i = 0; i <= n; i++)
        {
            a[i].x /= n;
        }
    }
    return;
}
int main()
{
    int n, m, k;
    scanf("%d%d%d", &n, &m, &k);
    n--;
    m--;
    scanf("%s%s", s, str);
    for(int i = 0; i < 4; i++)
    {
        for(int j = 0; j <= n; j++)
        {
            if(s[j] == ch[i])
            {
                d[std::max(0, j - k)]++;
                d[std::min(n + 1, j + k + 1)]--;
            }
        }
        int now = 0;
        for(int j = 0; j <= n; j++)
        {
            now += d[j];
            if(now)
            {
                f[j][i] = 1;
            }
        }
        memset(d, 0, sizeof(d));
    }
    int len = 2, lm = 1;
    while(len <= n + m)
    {
        len <<= 1;
        lm++;
    }
    for(int i = 1; i <= len; i++)
    {
        r[i] = (r[i >> 1] >> 1) | ((i & 1) << (lm - 1));
    }
    for(int i = 0; i < 4; i++)
    {
        for(int j = 0; j <= len; j++)
        {
            a[j] = b[j] = cp(0, 0);
        }
        for(int j = 0; j <= n; j++)
        {
            a[j].x = f[j][i];
        }
        for(int j = 0; j <= m; j++)
        {
            b[m - j].x = (int)(str[j] == ch[i]);
        }
        FFT(len, a, 1);
        FFT(len, b, 1);
        for(int j = 0; j <= len; j++)
        {
            a[j] = a[j] * b[j];
        }
        FFT(len, a, -1);
        for(int j = 0; j <= len; j++)
        {
            ans[j] += (int)(a[j].x + 0.5);
        }
    }
    int temp = 0;
    for(int i = m; i <= n; i++)
    {
        if(ans[i] == m + 1)
        {
            temp++;
        }
    }
    printf("%d
", temp);
    return 0;
}

代码乱了，用了CB的格式化，码风可能有点奇怪...