线性代数（矩阵乘法）：POJ 2778 DNA Sequence

  1 #include <iostream>
  2 #include <cstring>
  3 #include <cstdio>
  4 #include <queue>
  5 using namespace std;
  6 const int maxn=110;
  7 const int mod=100000;
  8 typedef unsigned long long ull;
  9 struct Matrix{
 10     int n;
 11     ull mat[maxn][maxn];
 12     Matrix(int n_,int on=0){
 13         n=n_;memset(mat,0,sizeof(mat));
 14         if(on)for(int i=1;i<=n;i++)mat[i][i]=1;
 15     }
 16     Matrix operator *(Matrix a){
 17         Matrix ret(n);
 18         unsigned long long l;
 19         for(int i=1;i<=n;i++)
 20             for(int k=1;k<=n;k++){
 21                 l=mat[i][k];
 22                 for(int j=1;j<=n;j++)
 23                     (ret.mat[i][j]+=l*a.mat[k][j]%mod)%=mod;    
 24             }
 25         return ret;    
 26     }
 27     Matrix operator ^(long long k){
 28         Matrix ret(n,1);
 29         while(k){
 30             if(k&1)
 31                 ret=ret**this;
 32             k>>=1;
 33             *this=*this**this;    
 34         }
 35         return ret;
 36     }
 37 };
 38 
 39 struct AC_automation{
 40     bool tag[maxn];
 41     int cnt,rt,ch[maxn][4],fail[maxn];
 42     AC_automation(){
 43         memset(tag,0,sizeof(tag));
 44         memset(fail,0,sizeof(fail));
 45         memset(ch,0,sizeof(ch));cnt=rt=1;
 46     }
 47     
 48     int ID(char c){
 49         if(c==‘A‘)return 0;
 50         else if(c==‘C‘)return 1;
 51         else if(c==‘G‘)return 2;
 52         else return 3;
 53     }
 54     
 55     void Insert(char *s){
 56         int len=strlen(s),p=rt;
 57         for(int i=0;i<len;i++)
 58             if(ch[p][ID(s[i])])
 59                 p=ch[p][ID(s[i])];
 60             else
 61                 p=ch[p][ID(s[i])]=++cnt;
 62         tag[p]=true;
 63     }
 64     
 65     void Build(){
 66         queue<int>q;
 67         for(int i=0;i<4;i++)
 68             if(ch[rt][i])
 69                 fail[ch[rt][i]]=rt,q.push(ch[rt][i]);
 70             else
 71                 ch[rt][i]=rt;
 72         
 73         while(!q.empty()){
 74             int x=q.front();q.pop();
 75             for(int i=0;i<4;i++)
 76                 if(ch[x][i]){
 77                     fail[ch[x][i]]=ch[fail[x]][i];
 78                     tag[ch[x][i]]|=tag[fail[ch[x][i]]];
 79                     q.push(ch[x][i]);
 80                 }
 81                 else
 82                     ch[x][i]=ch[fail[x]][i];
 83         }
 84     }
 85     
 86     void Solve(int k){
 87         Matrix A(cnt);    
 88         for(int i=1;i<=cnt;i++)
 89             for(int j=0;j<4;j++)
 90                 if(!tag[i]&&!tag[ch[i][j]])
 91                     A.mat[ch[i][j]][i]+=1;
 92         A=A^k;
 93         long long ans=0;
 94         for(int i=1;i<=cnt;i++)
 95             ans+=A.mat[i][1];
 96         printf("%lld\n",ans%mod);
 97     }
 98 }ac;
 99 char s[maxn];
100 
101 int main(){
102 #ifndef ONLINE_JUDGE
103     //freopen("","r",stdin);
104     //freopen("","w",stdout);
105 #endif
106     int tot,n;
107     scanf("%d%d",&tot,&n);
108     while(tot--){
109         scanf("%s",s);
110         ac.Insert(s);
111     }
112     ac.Build();
113     ac.Solve(n);
114     return 0;    
115 }