洛谷P3763 [TJOI2017]DNA（后缀自动机）

传送门

好像用SAM写的很少诶……

其实我一开始也没想到要用SAM的……主要是没有想到找的时候可以dfs……

首先建一个SAM，然后跑一遍dfs，枚举一下下一位，如果相同直接继续，否则就花费一次次数来改变它，保证改变次数小于等于3就行了

 // luogu-judger-enable-o2
 //minamoto
 #include<iostream>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 const int N=1e6+;
 int cnt[N<<],fa[N<<],ch[N<<][],l[N<<],last,tot;
 char s[N];int c[N],a[N],ans,m,val[N];
 void ins(int c){
     int p=last,np=++tot;last=np,l[np]=l[p]+,cnt[np]=;
     for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
     if(!p) fa[np]=;
     else{
         int q=ch[p][c];
         if(l[q]==l[p]+) fa[np]=q;
         else{
             int nq=++tot;l[nq]=l[p]+;
             memcpy(ch[nq],ch[q],sizeof(ch[q]));
             fa[nq]=fa[q],fa[q]=fa[np]=nq;
             for(;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
         }
     }
 }
 inline void calc(){
     memset(c,,sizeof(c));
     for(int i=;i<=tot;++i) ++c[l[i]];
     for(int i=;i<=tot;++i) c[i]+=c[i-];
     for(int i=;i<=tot;++i) a[c[l[i]]--]=i;
     for(int i=tot,p;i;--i)
     p=a[i],cnt[fa[p]]+=cnt[p];
 }
 void dfs(int p,int len,int j){
     if(len>m) return (void)(ans+=cnt[p]);
     for(int i=;i<;++i){
         if(!ch[p][i]) continue;
 //        val[s[len]]==i?dfs(ch[p][i],len+1,j):j<3?dfs(ch[p][i],len+1,j+1):(void)(1);
         if(val[s[len]]==i) dfs(ch[p][i],len+,j);
         else if(j<) dfs(ch[p][i],len+,j+);
     }
 }
 int main(){
 //    freopen("testdata.in","r",stdin);
     val['T']=,val['A']=,val['C']=,val['G']=;
     int T;scanf("%d",&T);
     while(T--){
         memset(ch,,sizeof(ch)),memset(cnt,,sizeof(cnt));
         last=tot=,ans=;
         scanf("%s",s+);int len=strlen(s+);
         for(int i=;i<=len;++i) ins(val[s[i]]);
         calc();
         scanf("%s",s+),m=strlen(s+);
         dfs(,,);printf("%d\n",ans);
     }
     return ;
 }