DNA Sequence POJ - 2778 AC自动机构造矩阵

构建一个长度为n的串s，不能包含模板串，问一共多少种s

把所有病毒节点和以病毒节点为后缀的节点标为病毒节点

从编号i的节点走到j时，如果i,j都不是病毒节点，这样的走法才是安全的，mat[i][j]++

其他就是一些矩阵的知识了

#include<bits/stdc++.h>
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<string.h>
#include<iostream>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
using namespace std;
#define ll long long
#define pb push_back
#define FOR(a) for(int i=1;i<=a;i++)
const int inf=0x3f3f3f3f;
const int maxn=1e2+9; 
const int mod=1e5;
const int sigma=4;

struct Matrix{
	ll mat[maxn][maxn];
	int n;
	Matrix(){};
	Matrix(int n1){n=n1;for(int i=0;i<n;i++)for(int j=0;j<n;j++)mat[i][j]=0;}
	Matrix operator *(const Matrix&b)const{
		Matrix ret=Matrix(n);
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				for(int k=0;k<n;k++){
					ret.mat[i][j]+=mat[i][k]*b.mat[k][j];
				}
				ret.mat[i][j]%=mod;
			}
		}
		return ret;
	}
};
Matrix matpow(Matrix mat,int n){
	Matrix ret=Matrix(mat.n);
	for(int i=0;i<mat.n;i++)ret.mat[i][i]=1;
	while(n){
		if(n&1)ret=ret*mat;
		mat=mat*mat;	n=n>>1;
	}
	return ret;
}
int getindex(char x){
	if(x=='A')return 0;if(x=='T')return 1;
	if(x=='C')return 2;return 3;
}

struct automata{
	int ch[maxn][sigma];
	int val[maxn];
	int f[maxn];
	int sz;

	int newnode(){
		memset(ch[sz],0,sizeof(ch[sz]));
		f[sz]=val[sz]=0;
		return sz++;
	}

	void init(){
		memset(val,0,sizeof(val));
		sz=0;
		newnode();
	}

	void insert(char *s,int v){	
		int u=0;
		int len=strlen(s);
		for(int i=0;i<len;i++){
			int id=getindex(s[i]);//s[i]-'a';
			if(!ch[u][id])ch[u][id]=newnode();
			u=ch[u][id];
		}
		val[u]++;
	}

	void build(){
		queue<int>q;
		q.push(0);	
		while(!q.empty()){
			int u=q.front();q.pop();

			if(val[f[u]]){	//当前字符串后缀是病毒，那这个也是病毒
							//因为更新时先标浅层的单纯病毒串
				val[u]++;
			}

			for(int i=0;i<sigma;i++){
				int v=ch[u][i];
				if(!v)ch[u][i]=ch[f[u]][i];
				else q.push(v);
				if(u&&v)f[v]=ch[f[u]][i];
			}
		}
	}
	
	void buildmat(Matrix &mat){
		for(int i=0;i<sz;i++)
			for(int j=0;j<sigma;j++)
				if(!val[i] && !val[ch[i][j]])
					mat.mat[i][ch[i][j]]++;
	}
}ac;

char s[20];
int main(){
	int n,m;
	while(~scanf("%d%d",&m,&n)){
		Matrix mat=Matrix(50);
		ac.init();
		for(int i=0;i<m;i++){
			scanf("%s",s);
			ac.insert(s,0);
		}
		ac.build();
		ac.buildmat(mat);
		mat=matpow(mat,n);
		ll ans=0;
		for(int i=0;i<ac.sz;i++){
			ans=(ans+mat.mat[0][i])%mod;
		}
		printf("%lld\n",ans);
		
	}
}