D. Fuzzy Search

标签: fft
11人阅读 评论(1) 收藏 举报
分类:

fft下标一定要从0开始,不然特麻烦
观察题目容易发现,可以在O(n)时间内判断对于位置i是否能在[i-k,i+k]区间存在AGCT
然后得到一个O(nm)的算法,考虑优化
那么分开考虑AGCT,令s[i]表示在第i个位置匹配到了多少字母。
s[i]=j=1mk[ij]t[j]其中k[i]表示第i个位置是否能匹配到某一个字母
类似于某个字符串匹配的题一样 ,t数组还是要取反。
然后fft即可,注意要四舍五入,复杂度O(nlogn)
c++代码如下:

#include<bits/stdc++.h>
#define PI acos(-1)
#define rep(i,x,y) for(register int i = x; i <= y; ++ i)
#define repd(i,x,y) for(register int i = x; i >= y; -- i)
#define id(x) (x == 'A' ? 0 : x == 'G' ? 1 : x == 'C' ? 2 : 3) 
using namespace std;
typedef long long ll;

const int N = 8e5+50;
char s[N],t[N];
int n,m,k,L,len,R[N],ans[N],lst;

struct cpx
{
    double x,y;
    cpx(){}
    cpx(double a,double b) { x = a,y = b; }
    inline void clear(){ x = 0,y = 0; }
    inline cpx operator * (cpx b) { return cpx(x*b.x - y * b.y,b.x*y + b.y * x); }  
    inline cpx operator *= (cpx b) { *this = *this * b; }  
    inline cpx operator + (cpx b) { return cpx(x + b.x,y + b.y); }  
    inline cpx operator - (cpx b) { return cpx(x - b.x,y - b.y); }  
}a[N],b[N];

inline void fft(cpx*a,int f)
{
    rep(i,0,len-1) if(i < R[i]) swap(a[i],a[R[i]]);
    for(register int i = 1;i < len; i <<= 1)
    {
        cpx wn = cpx(cos(PI/i),f*sin(PI/i));
        for(register int j = 0; j < len; j += i << 1)
        {
            cpx w = cpx(1,0); 
            for(register int k = 0; k < i; ++ k,w *= wn)
            {
                cpx x = a[j + k],y = w * a[i + j + k] ;
                a[j + k] = x + y;
                a[i + j + k] = x - y;
            }
        }
    }
    if(f == -1) rep(i,0,len-1) a[i].x /= len;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    scanf("%s",s);scanf("%s",t);

    for(int i = 0,j = m-1;i < j ;++i,--j) swap(t[i],t[j]);
    for(len = 1; len <= n + m; len <<= 1) ++ L;
    rep(i,0,len-1) R[i] = ((R[i>>1]>>1) | ((i&1) << (L - 1)));

    rep(i,0,3)
    {
        lst = -N;
        rep(j,0,len-1) a[j].clear(),b[j].clear();
        rep(j,0,k-1) if(i == id(s[j])) lst = j;
        rep(j,0,n-1)
        {
            if(i == id(s[j + k]) && j + k < n) lst = j+k;
            a[j].x = abs(lst - j) <= k;
        }

        rep(j,0,m-1) b[j].x = id(t[j]) == i;

        fft(a,1); fft(b,1);
        rep(j,0,len-1) a[j] *= b[j];
        fft(a,-1);
        rep(j,0,len-1) ans[j] += (int)(a[j].x+0.5);
    }

    int cnt = 0;
    rep(i,m-1,n-1) if(ans[i] == m) ++cnt;
    cout << cnt << endl;

    return 0;
}
查看评论

【FFT】 Codeforces Round #296 (Div. 1) D - Fuzzy Search

先将t串反转,然后对每一个字母做多项式乘法。。。全加起来就是t串匹配的最大字母数。。。 #include #include #include #include #include #incl...
  • blankcqk
  • blankcqk
  • 2015-03-19 16:41:44
  • 760

CodeForces 528 D.Fuzzy Search(FFT)

Description 给出一个长度为nn的串SS和一个长度为mm的串TT,两个串均只由′A′,′G′,′C′,′T′'A','G','C','T'组成,定义TT和SS在ii位置匹配为对于任意1≤j...
  • V5ZSQ
  • V5ZSQ
  • 2017-08-16 18:02:02
  • 222

codeforces 528D. Fuzzy Search (FFT优化DP)

题目描述传送门题目大意:给出一个母串和一个模板串,求模板串在母串中的匹配次数。 匹配时,如果用s[i]匹配t[j],那么只要s[i-k]-s[i+k]中有字母与t[j]相同即可算作匹配成功。其中s[...
  • clover_hxy
  • clover_hxy
  • 2017-02-23 07:57:59
  • 489

CodeForces 528D Fuzzy Search(FFT)

CodeForces 528D Fuzzy Search原题地址:http://codeforces.com/problemset/problem/528/D题意: 给给定两个字符串 S,T(字符集...
  • Bfk_zr
  • Bfk_zr
  • 2018-02-05 16:48:18
  • 100

Codeforces Round #296 (Div. 1) D. Fuzzy Search

D. Fuzzy Search time limit per test 3 seconds memory limit per test 256 megabytes i...
  • wufufuhome
  • wufufuhome
  • 2015-04-15 20:05:49
  • 153

【codeforces】528D. Fuzzy Search【FFT】

传送门:【codeforces】528D. Fuzzy Search题目分析:首先,我们先来了解一下FFTFFT求字符串匹配的方法。问题:求BB串在AA串中匹配的次数。解法: 我们用aia_i表示串...
  • u013368721
  • u013368721
  • 2015-05-07 19:55:48
  • 1732

Fuzzy/Partial Search in MongoDB

http://stackoverflow.com/a/27984038/2177408 The text search feature in MongoDB (as at 2.6) ...
  • SalmonellaVaccine
  • SalmonellaVaccine
  • 2016-12-25 08:16:35
  • 185

CF528D Fuzzy Search

Fuzzy Search 题目描述 传送门:http://codeforces.com/problemset/problem/528/D 题解 将AGTC分别考虑 考虑A字符,将S串...
  • wcy_1122
  • wcy_1122
  • 2018-02-01 00:47:10
  • 102

codeforces 528D Fuzzy Search

FFT
  • chai_jing
  • chai_jing
  • 2017-05-31 21:00:37
  • 184
    个人资料
    专栏达人 持之以恒
    等级:
    访问量: 2225
    积分: 889
    排名: 5万+
    博客专栏
    文章分类
    文章存档