CSP202009T5密信与计数

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
char s[10001];
char dit[60][60];
int Map[60][26],nex[60][26],len[10010];//Map[页数][明文]=密文,nex[页数][明文]=跳转页数 
long long f[1011][51][51],ans[100010];
int n,m,cnt=1,mod=998244353; 
struct node
{
	int son[26];
	int flag,fail;
}trie[250];
void insert(char* s)
{
	int u=1,len=strlen(s);
	for(int i=0;i<len;i++)
	{
		int v=s[i]-'a';
		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<26;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<26;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);					
		}
	}
}
int main()
{
	
	cin>>n>>m;
	for(int i=0;i<26;i++) //二十六行密文对应的明文需要处理 
	{
		for(int j=1;j<=n;j++)
		{
			cin>>s;
			Map[j][s[0]-'a']=i; 
			int l=strlen(s),v=0;
			if(l==2)v=s[1]-'0';  //提取出数字,表示跳转页数 
			else v=(s[1]-'0')*10+s[2]-'0';
			nex[j][s[0]-'a']=v;             
		}
	}
	int K=0;
	while(scanf("%s",dit[K])!=EOF)   //输入词典
	{	
		insert(dit[K]);
		len[K]=strlen(dit[K]);         //AC自动机的模板串的建立
		for(int i=0;i<len[K];i++)
		dit[K][i]=dit[K][i] - 'a';
		K++;
	}
	getFail();
	f[0][1][1]=1;
	for(int i=0;i<=m;i++)//长度 
	for(int j=1;j<=cnt;j++)//AC机上结点 
	for(int k=1;k<=n;k++)//页数 
	{
		if(f[i][j][k])
		{
			f[i][j][k]=f[i][j][k]%mod;
			ans[i]=ans[i]+f[i][j][k];
			bool flag=false;
			int tj,tk,x;
			for(int l=0;l<K;l++)//字典里的句子 
			{
				if(i+len[l]>m)continue;//如果该句子大于m就跳过
				flag=1;tj=j;tk=k;
				for(int p=0;p<len[l];p++) //句子的单个字符 
				{
					x=Map[tk][dit[l][p]];  //对应的密文 
					tk=nex[tk][dit[l][p]]; //跳转页数
					tj=trie[tj].son[x]; 
					if(trie[tj].flag){flag=0;break;}
					//加上的密文不包含词典中的任何一个串为子串 
				} 
				if(flag)
				f[i+len[l]][tj][tk]+=f[i][j][k];
			}
		}
	}
	for(int i=1;i<=m;i++)
	{
		ans[i]=ans[i]%mod;
		cout<<ans[i]<<endl;
	}
	return 0;
}