【POJ】2778 DNA Sequence（AC自动机+矩阵快速幂）

题目

传送门：QWQ

分析

对着Trie图搞快速幂。

为什么这样是对的呢？

详见：http://www.matrix67.com/blog/archives/276

有些地方还不是很理解。。。。。。为什么节点还要往下扩展？

有空再来搞一搞

代码

 //#include <bits/stdc++.h>
 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 using namespace std;
 typedef long long ll;
 const int maxn=, MOD=;
 int son[maxn][], fail[maxn], end[maxn],q[maxn] , newp, flag[maxn];
 char s[maxn];
 struct Matrix{
     ll m[maxn][maxn];
     Matrix(){memset(m,,sizeof(m));}
 };
 Matrix operator*(const Matrix &mat1,const Matrix &mat2){
     Matrix m;
     for(int i=;i<=newp;i++)
         for(int j=;j<=newp;j++)
             for(int k=;k<=newp;k++){
                 m.m[i][j]=(m.m[i][j]+mat1.m[i][k]*mat2.m[k][j])%MOD;
             }
     return m;
 }
 int find(int cur,int i){
     if(!cur) return ;
     if(son[cur][i]) return son[cur][i];
     return find(fail[cur],i);
 }
 int main(){
     int n,L;
     while(scanf("%d%d",&n,&L)==){
         int l,r;
         memset(son,,sizeof(son)); memset(fail,,sizeof(fail)); memset(end,,sizeof(end)); memset(q,,sizeof(q));
         newp=;
         for(int i=;i<=n;i++){
             scanf("%s",s+);
             int cur=,len=strlen(s+);
             for(int j=;j<=len;j++){
                 if(!son[cur][s[j]-'A']) son[cur][s[j]-'A']=++newp;
                 cur=son[cur][s[j]-'A'];
             }
             end[cur]=i; flag[cur]=;
         }

         Matrix ans,a;
         for(q[l=r=]=;l<=r;l++){
             for(int i=;i<;i++){
                 if(son[q[l]][i]){
                     fail[son[q[l]][i]]=find(fail[q[l]],i);
                     q[++r]=son[q[l]][i];
                 }
                 else son[q[l]][i]=find(fail[q[l]],i);
                 flag[son[q[l]][i]]|=flag[fail[son[q[l]][i]]];
             }
         }

         for(int i=;i<=newp;i++){
             if(flag[i]) continue;
             for(int j=;j<;j++){
                 int q=j+'A';
                 if(((q=='A'||q=='C'||q=='G'||q=='T'))&&flag[son[i][j]]==){
                     a.m[i][max(son[i][j],)]++;

                 }
             }
         }
         for(int i=;i<=newp;i++,puts(""))
             for(int j=;j<=newp;j++){
                 printf("%lld ",a.m[i][j]);
             }
         for(int i=;i<=newp;i++) ans.m[i][i]=;
         while(L){
             if(L&) ans=ans*a;
             a=a*a;
             L>>=;
         }
         ll res=;
         for(int i=;i<=newp;i++) res=(res+ans.m[][i])%MOD;
         printf("%d\n",res);
     }
 }
 /*
 5 10
 ACGE
 CG
 CCGG
 ACG
 TTAC
 */