1. 程式人生 > >bzoj1559:[JSOI2009]密碼

bzoj1559:[JSOI2009]密碼

string max 當前 \n %s pty 兩種 char s scanf

沒臉扔傳送門了
寫個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);
}

bzoj1559:[JSOI2009]密碼