码迷,mamicode.com
首页 > 其他好文 > 详细

[CF528D]Fuzzy Search

时间:2018-04-06 22:36:47      阅读:297      评论:0      收藏:0      [点我收藏+]

标签:org   amp   tor   求和   inline   AC   space   cst   lin   

luogu

sol

这种字符串匹配的问题显然可以把一个串\(reverse\)过来然后用\(FFT\)做吧。
对每种字母分开考虑,设两个多项式\(A(x),B(x)\),其中
\[A(x)=\sum_{i=0}^{n-1}[区间[i-k,i+k]内存在该种字符]x^i\]
\[B(x)=\sum_{i=0}^{n-1}[t[i]为该种字符]x^i\]
然后两个多项式卷一下就是这种字符在第\(i\)个位置上的匹配吧。
最后对每次的系数求和,若恰好等于\(|T|\)则说明匹配成功。
复杂度\(O(4n\log n)\)

code

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int _ = 8e5+5;
const double Pi = acos(-1);
struct Complex{
    double rl,im;
    Complex(){rl=im=0;}
    Complex(double a,double b){rl=a,im=b;}
    Complex operator + (Complex b)
        {return Complex(rl+b.rl,im+b.im);}
    Complex operator - (Complex b)
        {return Complex(rl-b.rl,im-b.im);}
    Complex operator * (Complex b)
        {return Complex(rl*b.rl-im*b.im,rl*b.im+im*b.rl);}
}w[_],a[_],b[_];
int N,n,m,k,rev[_],l,tmp[_],ans[_],Ans;
char s[_],t[_],fg[4]={'A','G','C','T'};
void FFT(Complex *P,int opt)
{
    for (int i=0;i<N;++i) if (i<rev[i]) swap(P[i],P[rev[i]]);
    for (int i=1;i<N;i<<=1)
        for (int p=i<<1,j=0;j<N;j+=p)
            for (int k=0;k<i;++k)
            {
                Complex W=w[N/i*k];W.im*=opt;
                Complex X=P[j+k],Y=W*P[j+k+i];
                P[j+k]=X+Y;P[j+k+i]=X-Y;
            }
    if (opt==-1) for (int i=0;i<N;++i) P[i].rl/=N;
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);scanf("%s%s",s,t);reverse(t,t+m);
    for (N=1;N<n+m;N<<=1) ++l;--l;
    for (int i=0;i<N;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l);
    for (int i=0;i<N;++i) w[i]=Complex(cos(Pi*i/N),sin(Pi*i/N));
    for (int zsy=0;zsy<4;++zsy)
    {
        for (int i=0;i<N;++i) a[i].rl=a[i].im=b[i].rl=b[i].im=0;
        for (int i=0,pos=-1e9;i<n;++i)
        {
            if (s[i]==fg[zsy]) pos=i;
            if (pos>=i-k) a[i].rl=1;
        }
        for (int i=n-1,pos=1e9;~i;--i)
        {
            if (s[i]==fg[zsy]) pos=i;
            if (pos<=i+k) a[i].rl=1;
        }
        for (int i=0;i<m;++i) if (t[i]==fg[zsy]) b[i].rl=1;
        FFT(a,1);FFT(b,1);
        for (int i=0;i<N;++i) a[i]=a[i]*b[i];
        FFT(a,-1);
        for (int i=0;i<n;++i) ans[i]+=(int)(a[i].rl+0.5);
    }
    for (int i=0;i<n;++i) if (ans[i]==m) ++Ans;
    printf("%d\n",Ans);return 0;
}

[CF528D]Fuzzy Search

标签:org   amp   tor   求和   inline   AC   space   cst   lin   

原文地址:https://www.cnblogs.com/zhoushuyu/p/8728888.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!