poj 2778 AC自动机+矩阵乘法

首先利用病毒串建立tire图，没个节点代表某种后缀，然后利用AC自动机对各个后缀进行转移，得到一个矩阵，然后利用快速幂求解矩阵连乘。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
using namespace std;
const int nMax=105;
const int S_NUM=4;
const int N_NUM=105;
const int M=100000;
struct Tree
{
    int index,flag;
    Tree *fail,*next[S_NUM];
}t[nMax],*Q[nMax];
struct Mat
{
	long long mat[N_NUM][N_NUM];
	int n,m;
	void init(int r,int c)
	{
		n=r;m=c;
	}
	void init_e()
	{
		for(int i=0;i<n;i++)
			for(int j=0;j<m;j++)
				mat[i][j]=(i==j);
	}
};
int cnt,L,C;
long long a[N_NUM][N_NUM];
char key[20];
Tree *root;
int get(char c)
{
   if(c=='A') return 0;
   else if(c=='G') return 1;
   else if(c=='C') return 2;
   else return 3;
}
Tree* new_node()
{
    Tree *p=&t[cnt];
    p->index=cnt++;
    p->flag=0;
    memset(p->next,0,sizeof(p->next));
    return p;
}
void insert(char s[])
{
    Tree *p=root;
    for(int i=0;s[i];i++)
    {
        int index=get(s[i]);
        if(p->next[index]==NULL)
            p->next[index]=new_node();
        p=p->next[index];
    }
    p->flag=1;
}
void build_ac()
{
    int f=0,r=-1;
    Q[++r]=root;
    while(f<=r)
    {
        Tree *p=Q[f++];
        for(int i=0;i<S_NUM;i++)
        {
            if(p->next[i]!=NULL)
            {
                if(p==root) p->next[i]->fail=root;
                else
                {
                    p->next[i]->fail=p->fail->next[i];
                    if(p->fail->next[i]->flag!=0)
                        p->next[i]->flag=1;
                }
                Q[++r]=p->next[i];
            }
            else
            {
                if(p==root) p->next[i]=root;
                else p->next[i]=p->fail->next[i];
            }
            if(p->next[i]->flag==0) a[p->index][p->next[i]->index]++;
        }
    }
}
Mat Mul(Mat a,Mat b)
{
	Mat ret;
	ret.init(a.n,b.m);
	memset(ret.mat,0,sizeof(ret.mat));
	for(int i=0;i<a.n;i++)
		for(int j=0;j<b.m;j++)
			for(int k=0;k<b.n;k++)
			{
				ret.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
				if(ret.mat[i][j]>M) ret.mat[i][j]%=M;
			}
	return ret;
}
Mat exp(Mat a,int k)
{
	Mat ret=a,tmp=a;
	ret.init_e();
	for(;k;k>>=1)
	{
		if(k&1)
		{
			ret=Mul(ret,tmp);
		}
		tmp=Mul(tmp,tmp);
	}
	return ret;
}
int main()
{
//    freopen("test.txt","r",stdin);
    scanf("%d%d",&L,&C);
    cnt=0;root=new_node();
    for(int i=0;i<L;i++)
    {
        scanf("%s",key);
        insert(key);
    }
    build_ac();
    Mat b; b.init(cnt,cnt);
    memcpy(b.mat,a,sizeof(a));
    Mat ans=exp(b,C);
    int res=0;
    for(int i=0;i<cnt;i++)
        res=(res+ans.mat[0][i])%M;
    printf("%d\n",res);
    return 0;

}