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 = ;
 const double pi = 3.1415926535897932384626;
 const char ch[] = {'A', 'C', 'G', 'T'};
 struct cp
 {
     double x, y;
     cp(double X = , double Y = )
     {
         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 << ], b[N << ];
 int r[N << ], ans[N << ], f[N][], d[N];
 char s[N], str[N];
 inline void FFT(int n, cp *a, int f)
 {
     for(int i = ; i < n; i++)
     {
         if(i < r[i])
         {
             std::swap(a[i], a[r[i]]);
         }
     }
     for(int len = ; len < n; len <<= )
     {
         cp Wn(cos(pi / len), f * sin(pi / len));
         for(int i = ; i < n; i += (len << ))
         {
             cp w(, );
             for(int j = ; 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 == -)
     {
         for(int i = ; 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 = ; i < ; i++)
     {
         for(int j = ; j <= n; j++)
         {
             if(s[j] == ch[i])
             {
                 d[std::max(, j - k)]++;
                 d[std::min(n + , j + k + )]--;
             }
         }
         int now = ;
         for(int j = ; j <= n; j++)
         {
             now += d[j];
             if(now)
             {
                 f[j][i] = ;
             }
         }
         memset(d, , sizeof(d));
     }
     int len = , lm = ;
     while(len <= n + m)
     {
         len <<= ;
         lm++;
     }
     for(int i = ; i <= len; i++)
     {
         r[i] = (r[i >> ] >> ) | ((i & ) << (lm - ));
     }
     for(int i = ; i < ; i++)
     {
         for(int j = ; j <= len; j++)
         {
             a[j] = b[j] = cp(, );
         }
         for(int j = ; j <= n; j++)
         {
             a[j].x = f[j][i];
         }
         for(int j = ; j <= m; j++)
         {
             b[m - j].x = (int)(str[j] == ch[i]);
         }
         FFT(len, a, );
         FFT(len, b, );
         for(int j = ; j <= len; j++)
         {
             a[j] = a[j] * b[j];
         }
         FFT(len, a, -);
         for(int j = ; j <= len; j++)
         {
             ans[j] += (int)(a[j].x + 0.5);
         }
     }
     int temp = ;
     for(int i = m; i <= n; i++)
     {
         if(ans[i] == m + )
         {
             temp++;
         }
     }
     printf("%d\n", temp);
     return ;
 }

AC代码

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