BZOJ1559: [JSOI2009]密码

https://lydsy.com/JudgeOnline/problem.php?id=1559

分析:

设(f[i][j][s])表示已经有了(i)位，在ac自动机上(j)号结点上，出现子串状态为(s)的方案数。
输出方案也好做，记忆化+标记，然后直接搜，这样复杂度就等于答案串个数乘串长了。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <cmath>
using namespace std;
typedef long long ll;
#define N 105
char w[25];
int ch[N][26],cnt=1,fail[N],K,n,sta[N],Q[N],mask;
ll f[30][N][1050];
char is[30][N][1050];
char ans[N];
void insert(int id) {
	int i,p=1,m=strlen(w+1);
	for(i=1;i<=m;i++) {
		int &k=ch[p][w[i]-'a'];
		if(!k) k=++cnt;
		p=k;
	}
	sta[p]|=(1<<(id-1));
}
void build() {
	int i,p;
	for(i=0;i<26;i++) ch[0][i]=1;
	int l=0,r=0; Q[r++]=1;
	while(l<r) {
		p=Q[l++]; sta[p]|=sta[fail[p]];
		for(i=0;i<26;i++) {
			if(ch[p][i]) fail[ch[p][i]]=ch[fail[p]][i],Q[r++]=ch[p][i];
			else ch[p][i]=ch[fail[p]][i];
		}
	}
}
int dfs(int nd,int p,int st) {
	if(is[nd][p][st]!=-1) return is[nd][p][st];
	if(nd==n) {
		if(st==mask) return is[nd][p][st]=1;
		else return is[nd][p][st]=0;
	}
	int re=0;
	int i;
	for(i=0;i<26;i++) re|=dfs(nd+1,ch[p][i],st|sta[ch[p][i]]);
	return is[nd][p][st]=re;
}
void solve(int nd,int p,int st) {
	if(is[nd][p][st]!=1) return ;
	int i;
	if(nd==n) {
		if(st==mask) printf("%s
",ans+1);
		return;
	}
	for(i=0;i<26;i++) {
		if(is[nd+1][ch[p][i]][st|sta[ch[p][i]]]==1) {
			//printf("%d:%c,f=%lld
",nd+1,i+'a',f[nd+1][ch[p][i]][st|sta[ch[p][i]]]);
			ans[nd+1]=i+'a';
			solve(nd+1,ch[p][i],st|sta[ch[p][i]]);
		}
	}
}
int main() {
	//freopen("fire.in","r",stdin);
	//freopen("fire.out","w",stdout);
	scanf("%d%d",&n,&K);
	int i;
	mask=(1<<K)-1;
	for(i=1;i<=K;i++) {
		scanf("%s",w+1);
		insert(i);
	}
	build();
	f[0][1][0]=1;
	int j,k,l;
	for(i=0;i<n;i++) {
		for(j=1;j<=cnt;j++) {
			for(k=0;k<=mask;k++) if(f[i][j][k]) {
				for(l=0;l<26;l++) {
					int p=ch[j][l];
					f[i+1][p][k|sta[p]]+=f[i][j][k];
				}
			}
		}
	}
	ll ans=0;
	for(i=1;i<=cnt;i++) ans+=f[n][i][mask];
	printf("%lld
",ans);
	if(ans>42) return 0;
	memset(is,-1,sizeof(is));
	dfs(0,1,0);
	solve(0,1,0);
}