洛谷3763：[TJOI2017]DNA——题解

https://www.luogu.org/problemnew/show/P3763

加里敦大学的生物研究所，发现了决定人喜不喜欢吃藕的基因序列S,有这个序列的碱基序列就会表现出喜欢吃藕的性状，但是研究人员发现对碱基序列S,任意修改其中不超过3个碱基，依然能够表现出吃藕的性状。现在研究人员想知道这个基因在DNA链S0上的位置。所以你需要统计在一个表现出吃藕性状的人的DNA序列S0上，有多少个连续子串可能是该基因，即有多少个S0的连续子串修改小于等于三个字母能够变成S。

是的这篇代码过不了BZOJ（因为懒得卡了/不想写SAM/不会DC3……众多原因）

其实容错三次匹配并不吓人，我们可以先跳跃匹配到匹配不上的地方，然后cnt++，继续跳跃……直到匹配完全或者cnt>3为止。

这个跳跃完全可以枚举起点，然后用SA来求lcp进而实现跳跃匹配以此变成$O(n)$的。

所以总复杂度是$O(Tnlogn)$的……只要卡卡就能过洛谷。

当然为了过BZOJ，要么常数优秀（写SAM，然后遍历，每次选择一个节点往下走，如果和当前节点匹配不上则cnt++，匹配复杂度不变但是常数小），要么就学DC3，要么……其实后缀数组卡卡也能过。

#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=2e5+;
inline int turn(char ch){
    if(ch==)return ;
    if(ch=='A')return ;
    if(ch=='G')return ;
    if(ch=='C')return ;
    if(ch=='T')return ;
    return ;
}
char s[N],p[N];
int n,rk[N],height[N],w[N],sa[N];
inline bool pan(int *x,int i,int j,int k){
    int ti=i+k<n?x[i+k]:-;
    int tj=j+k<n?x[j+k]:-;
    return ti==tj&&x[i]==x[j];
}
void SA_init(){
    int *x=rk,*y=height,r=;
    for(int i=;i<r;i++)w[i]=;
    for(int i=;i<n;i++)w[turn(s[i])]++;
    for(int i=;i<r;i++)w[i]+=w[i-];
    for(int i=n-;i>=;i--)sa[--w[turn(s[i])]]=i;
    r=;x[sa[]]=;
    for(int i=;i<n;i++)
        x[sa[i]]=s[sa[i]]==s[sa[i-]]?r-:r++;
    for(int k=;r<n;k<<=){
        int yn=;
        for(int i=n-k;i<n;i++)y[yn++]=i;
        for(int i=;i<n;i++)
            if(sa[i]>=k)y[yn++]=sa[i]-k;
        for(int i=;i<r;i++)w[i]=;
        for(int i=;i<n;i++)w[x[y[i]]]++;
        for(int i=;i<r;i++)w[i]+=w[i-];
        for(int i=n-;i>=;i--)sa[--w[x[y[i]]]]=y[i];
        swap(x,y);r=;x[sa[]]=;
        for(int i=;i<n;i++)
            x[sa[i]]=pan(y,sa[i],sa[i-],k)?r-:r++;
    }
}
void height_init(){
    int i,j,k=;
    for(int i=;i<=n;i++)rk[sa[i]]=i;
    for(int i=;i<n;i++){
        if(k)k--;
        int j=sa[rk[i]-];
        while(s[i+k]==s[j+k])k++;
        height[rk[i]]=k;
    }
}
int f[N][],lg[N];
inline int qpow(int a){return <<a;}
void st_init(){
    for(int i=;i<=n;i++){
        f[i-][]=height[i];
        lg[i]=lg[i-];
        if((<<lg[i]+)==i)lg[i]++;
    }
    for(int j=;j<=lg[n];j++){
        for(int i=;i<n;i++){
            if(i+qpow(j)->=n)break;
            f[i][j]=min(f[i][j-],f[i+qpow(j-)][j-]);
        }
    }
}
int lcp(int i,int j){
    int l=rk[i],r=rk[j];if(l>r)swap(l,r);
    l--;r--;if(r<)return ;l++;
    int len=r-l+,k=lg[len],h=qpow(k);
    return min(f[l][k],f[r-h+][k]);
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        scanf("%s%s",s,p);
        n=strlen(s);int m=strlen(p);
        s[n++]='#';
        for(int i=;i<m;i++)s[n++]=p[i];
        s[n++]=;SA_init();n--;height_init();st_init();
        int ans=;
        for(int i=;i<n-*m;i++){
            int cnt=;
            for(int j=;j<m&&cnt<=;){
                if(s[i+j]!=s[n-m+j])cnt++,j++;
                else j+=lcp(i+j,n-m+j);
            }
            if(cnt<=)ans++;
        }
        printf("%d\n",ans);
    }
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++