POJ 2778 DNA Sequence

题意：有m种DNA序列是有疾病的，问有多少种长度为n的DNA序列不包含任何一种有疾病的DNA序列。

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
struct node
{
	int son[26];
	int flag,fail;
}trie[250];
int mod=100000;
char s[200001];
long long n,m,cnt=1;
struct matrix
{
    long long s[150][150];
    matrix()
	{
        memset(s,0,sizeof s);
    }
}A;
matrix operator *(const matrix &a,const matrix &b)
{
    matrix c;
    for(int i=1;i<=cnt;i++)
	for(int j=1;j<=cnt;j++)
	for(int k=1;k<=cnt;k++)  //max则初始化-9999999；加则初始化为0 
	c.s[i][j]=(a.s[i][k]*b.s[k][j]+c.s[i][j])%mod;
    return c;
}
matrix pow(matrix a,long long p)
{
	matrix t;
	for(int i=1;i<=cnt;i++)t.s[i][i]=1;	
	while(p)
	{
		if(p&1)t=t*a;
		a=a*a;
		p>>=1;
	}
	return t;
}
void insert(char* s)
{
	int u=1,len=strlen(s),v;
	for(int i=0;i<len;i++)
	{
		if(s[i]=='A')v=0;
		if(s[i]=='C')v=1;
		if(s[i]=='G')v=2;
		if(s[i]=='T')v=3;
		if(!trie[u].son[v])trie[u].son[v]=++cnt;
		u=trie[u].son[v];
	}
	trie[u].flag=1;//标志结束 
}
queue <int> q;
void getFail()
{
	for(int i=0;i<4;i++)trie[0].son[i]=1;//初始化0的所有儿子都是1
	q.push(1);
	trie[1].fail=0;	//将根的fail为0 
	while(!q.empty())
	{
		int u=q.front();q.pop();
		if(trie[trie[u].fail].flag)trie[u].flag=1;  
		for(int i=0;i<4;i++)    //遍历所有儿子
		{
			int v=trie[u].son[i];		
			int Fail=trie[u].fail;			//trie[Fail].son[i]就是和v值相同的点
			if(!v){trie[u].son[i]=trie[Fail].son[i];continue;}	//不存在该节点
			trie[v].fail=trie[Fail].son[i];	
			q.push(v);					
		}
	}
}
void crmatrix()
{
	for(int i=1;i<=cnt;i++)
	for(int j=0;j<4;j++)
	{
		if(!trie[i].flag)
		{
			int p=trie[i].son[j];
			if(p>1)
			{
				if(!trie[p].flag)A.s[i][p]++;
			}
			else A.s[i][1]++;	
		}
	}
}
int main()
{
	while (~scanf("%d%d", &n, &m)) {
	trie[0].flag=0; 
	for(int i=1;i<=n;i++)
	{
		cin>>s;
		insert(s);
	}	
	getFail();
	crmatrix();
    // 2 1 1 1 6 4 5 3 2 1 1 1 2 1 1 1 2 1 1 1 6 4 5 3
    // 1 0 0 0 5 3 4 2 1 0 0 0 1 0 0 0 1 0 0 0 5 3 4 2
	A=pow(A,m);
	long long ans=0;
	for(int i=0;i<=cnt;i++)ans+=A.s[1][i];
	printf("%lld\n",ans%mod);
	}
	return 0;
}