bzoj1559:[JSOI2009]密码

没脸扔传送门了
写个AC自动机的题目被爆搜卡住,最终我决定不写这个爆搜了,太恶心了
状态压缩dp!
直接设\(f[i][j][S]为长度为i,当前为第j个节点,经过的密码状态为S\)
因为哪怕只有一个位置不确定都有两种位置可以填,一共就有26*2=52>42种情况
所以如果要输出方案,所有的密码串一定是能填满原串的,爆搜即可(处理一下重复的串吗,搜索处理一下前后缀相同)
然后几乎就没了(只不过我是因为太菜了,爆搜写不出)
给个AC自动机dp的代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<string>
using namespace std;
void read(int &x) {
	char ch; bool ok;
	for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
	for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=110;queue<int>q;long long f[26][maxn][(1<<10)+10],ans;
int fail[maxn],n,m,rt=1,id=1,ch[maxn][26],ed[110],tot;
struct oo{char s[15];int len;}w[12];
void insert(char *s,int d)
{
	w[d].len=strlen(s+1);rt=1;
	for(rg int i=1;i<=w[d].len;i++)
	{
		int now=w[d].s[i]-'a';
		if(!ch[rt][now])ch[rt][now]=++id;
		rt=ch[rt][now];
	}
	ed[rt]|=1<<(d-1);
}
void bfs()
{
	q.push(1);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(rg int i=0;i<26;i++)
		{
			if(!ch[x][i]){ch[x][i]=fail[x]?ch[fail[x]][i]:1;continue;}
			int j=fail[x],z=ch[x][i];q.push(z);
			while(j&&!ch[j][i])j=fail[j];
			fail[z]=j?ch[j][i]:1;
			ed[z]|=ed[fail[z]];
		}
	}
}
int main()
{
	read(n),read(m);
	for(rg int i=1;i<=m;i++)scanf("%s",w[i].s+1),insert(w[i].s,i);
	f[0][1][0]=1;int tot=(1<<m)-1;bfs();
	for(rg int i=1;i<=n;i++)
		for(rg int j=1;j<=id;j++)
			for(rg int k=0;k<=tot;k++)
				if(f[i-1][j][k])
					for(rg int t=0;t<26;t++)
						f[i][ch[j][t]][k|ed[ch[j][t]]]+=f[i-1][j][k];
	for(rg int i=1;i<=id;i++)ans+=f[n][i][tot];printf("%lld\n",ans);
}
posted @ 2019-02-27 20:25  蒟蒻--lichenxi  阅读(118)  评论(0编辑  收藏  举报