poj 2778

AC自动机+矩阵乘法。和多数人一样。也是看了Matrix67的博客。http://www.cnblogs.com/lj030/archive/2013/05/17/3083718.html

#include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <queue>
 #define LL long long
 using namespace std;
 const int maxn=11;
 const int inf=0x3f3f3f3f;
 const int mod=100000;
 int ch[12*12][4];
 int val[12*12],tot,f[12*12];
 struct Matrix
 {
     LL g[11*11][11*11];
 };
 Matrix operator * (Matrix a,Matrix b)
{
    int i,j,k;
    Matrix c;
    for(i=0;i<tot;i++)
    {
        for(j=0;j<tot;j++)
        {
            c.g[i][j]=0;
            for(k=0;k<tot;k++)
                c.g[i][j]=(c.g[i][j]+a.g[i][k]*b.g[k][j])%mod;
        }
    }
    return c;
}
 int getV(char c)
 {
     switch(c)
     {
     case 'A':return 0;
     case 'G':return 1;
     case 'C':return 2;
     case 'T':return 3;
     }
 }
 void Insert(char *s)
 {
     int u=0,len=strlen(s),i,tv;
     for(i=0;i<len;i++)
     {
         tv=getV(s[i]);
         if(!ch[u][tv])
         {
             memset(ch[tot],0,sizeof(ch[tot]));
             ch[u][tv]=tot++;
         }
         u=ch[u][tv];
     }
     val[u]=1;
 }
 void getFail()
 {
     queue<int> q;
     f[0]=0;
     int i,u;
     for(i=0;i<4;i++)
     {
         u=ch[0][i];
         if(u)
         {
             f[u]=0;
             q.push(u);
         }
     }
     int v,r;
     while(!q.empty())
     {
         r=q.front();q.pop();
         for(i=0;i<4;i++)
         {
             u=ch[r][i];
             if(!u) { ch[r][i]=ch[f[r]][i]; }
             else
             {
             if(val[r]) val[u]=1;
             q.push(u);
             v=f[r];
             while(v&&!ch[v][i]) v=f[v];
             f[u]=ch[v][i];
             if(val[f[u]]) val[u]=1;
             }
         }
     }
 }
 Matrix M_pow(Matrix tem,int n)
 {
     Matrix tem2;
     memset(tem2.g,0,sizeof(tem2.g));
     for(int i=0;i<tot;i++) tem2.g[i][i]=1;
     for(;n;n>>=1)
     {
         if(n&1) tem2=tem2*tem;
        tem=tem*tem;
     }
     return tem2;
 }
 int main()
 {
     int m,n;
     while(~scanf("%d%d",&m,&n))
     {
         memset(val,0,sizeof(val));
     int i,j;
     char s[12];
     tot=1;
     for(i=0;i<m;i++)
     {
         scanf("%s",s);
         Insert(s);
     }
     getFail();
     Matrix tem;
     for(i=0;i<tot;i++)
         for(j=0;j<tot;j++)
            tem.g[i][j]=0;
     int v;
    for(i=0;i<tot;i++)
    {
        for(j=0;j<4;j++)
        {
           v=ch[i][j];
           if(val[v]) continue;
           else
               tem.g[i][v]++;
        }
    }
    Matrix ans=M_pow(tem,n);
    int re=0;
    for(i=0;i<tot;i++) re=re+ans.g[0][i];
    printf("%d\n",re%mod);
     }
     return 0;
 }

---恢复内容结束---