LA 4126 Password Suspects AC自动机+dp

LA 4126

题意：已知某未知字符串长度为n，全部为小写，还知道该字符串的m个子串，求符合条件的该字符串个数，如果个数小于等于42，输出所有可能的字符串。

思路：将m个子串建立自动机，设d[ k ][ u ][ s ]为当前在节点u，已经走过的子串集合为S，已经走了k步的路径数，那么当k==n且s为全集的时候，就把合理的走法，把路径加上去，由于还要输出路径，因此把dp公式反过来，d[0][0][0]就是答案，当走到第n步时，如果集合没满，当前路径不可取，返回0，否则返回1，公式是d[ k ][ u ][ s ]+=d[ k+1 ][ son ][ s | val[ s ] ]。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long LL;
const int maxn=105;
int n,all;
char str[40];
struct AC
{
	int ch[maxn][26];
	int val[maxn];
	int f[maxn],last[maxn];
	int sz;
	LL d[26][maxn][1<<10];
	int vis[26][maxn][1<<10];
	void init()
	{
		sz=1;
		memset(ch[0],0,sizeof(ch[0]));
		memset(val,0,sizeof(val));
		memset(d,0,sizeof(d));
		memset(vis,0,sizeof(vis));
	}
	void insert(char *s,int v)
	{
		int u=0,len=strlen(s);
		for(int i=0;i<len;i++)
		{
			int j=s[i]-'a';
			if(!ch[u][j])
			{
				memset(ch[sz],0,sizeof(ch[sz]));
				val[sz]=0;
				ch[u][j]=sz++;
			}
			u=ch[u][j];
		}
		val[u]|=v;
	}
	void getfail()
	{
		int u=0;
		queue<int>q;
		f[0]=last[0]=0;
		for(int i=0;i<26;i++)
		if(ch[u][i])
		{
			q.push(ch[u][i]);
			f[ch[u][i]]=last[ch[u][i]]=0;
		}
		while(!q.empty())
		{
			int r=q.front();q.pop();
			for(int c=0;c<26;c++)
			{
				int u=ch[r][c];
				if(!u)
				{
					ch[r][c]=ch[f[r]][c];//这一步的作用是当某个字符串走完后可以继续走另一条字符串 
					continue;
				}
				q.push(u);
				int v=f[r];
				while(v&&!ch[r][c])v=f[v];
				f[u]=ch[v][c];
				last[u]=val[f[u]]?f[u]:last[f[u]];
				val[u]|=val[f[u]];//当某个字符串是另一个的前缀，走完长的那条字符串相当于走了两条 
			}
		}	
	}
	LL dfs(int k,int u,int s)
	{
		if(vis[k][u][s])return d[k][u][s];
		vis[k][u][s]=1;
		if(k==n)return d[k][u][s]=s==all?1:0;
		for(int i=0;i<26;i++)
		d[k][u][s]+=dfs(k+1,ch[u][i],s|val[ch[u][i]]);
		return d[k][u][s];
	}
	void print(int k,int u,int s)
	{
		if(k==n)
		{
			for(int i=1;i<=k;i++)
			printf("%c",str[i]);
			printf("\n");
			return;
		}
		for(int i=0;i<26;i++)
		{
			str[k+1]=i+'a';
			int v=ch[u][i];
			int st=s|val[v];
			if(d[k+1][v][st])
			print(k+1,v,st);
		}
	}
}ac;
int main()
{
	int m,kase=0;
	char s[25];
	while(~scanf("%d%d",&n,&m)&&n+m)
	{
		ac.init();
		for(int i=0;i<m;i++)
		{
			scanf("%s",s);
			ac.insert(s,1<<i);
		}
		ac.getfail();
		all=(1<<m)-1;
		LL ans=ac.dfs(0,0,0);
		printf("Case %d: %lld suspects\n",++kase,ans);
		if(ans<=42)
		ac.print(0,0,0);
	}
}