HDU - 2825 Wireless Password(AC自动机+状压DP)

题意:

给你m个字符串S,询问有多少种字符串满足长度为n且包含k个以上字符串S。

题解:

做了那么多了,这类状压DP太水了…

设dp[i][j][S]表示当前处理到第i个字符,在AC自动机上的节点标号为j,已经包含的字符串集合为S

然后循环转移就好了,确实太水了,也说不了更多的东西了…看代码吧…


AC代码:

#include
using namespace std;
const int MAXN = 110;
const int MOD = 20090717;
int nxt[MAXN][26],fail[MAXN],vis[MAXN];
int sta[MAXN],dp[30][110][(1<<10)+10],cnt[(1<<10)+10],tot;
char s[MAXN];
inline void Init(){
    for(int i=0;i<=tot;i++){
        sta[i]=fail[i]=vis[i]=0;
        for(int j=0;j<26;j++) nxt[i][j]=0;
    }
    tot=1;
}
inline void Insert(char *s,int k){
    int rt=0,len=strlen(s);
    for(int i=0;i<len;i++){
        if(!nxt[rt][s[i]-'a']) nxt[rt][s[i]-'a']=++tot;
        rt=nxt[rt][s[i]-'a'];
    }
    sta[rt] |= (1<<k);
}
inline void Build(){
    queue<int> que;
    for(int i=0;i<26;i++) if(nxt[0][i]) que.push(nxt[0][i]);
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=0;i<26;i++){
            if(nxt[u][i]){
                fail[nxt[u][i]] = nxt[fail[u]][i];
                que.push(nxt[u][i]);
                sta[nxt[u][i]] |= sta[nxt[fail[u]][i]];
            }else{
                nxt[u][i] = nxt[fail[u]][i];
            }
        }
    }
}
inline void add(int &x,int y){ x = (x+y)%MOD; }
inline int f(int x){
    int res=0;
    for(int i=0;i<10;i++) if((x>>i)&1) res++;
    return res;
}
inline void pre(){
    for(int i=0;i<(1<<10);i++) cnt[i]=f(i);
}
int main(){
    pre();
    int n,m,k; while(~scanf("%d%d%d",&n,&m,&k)){
        if(n==0 && m==0 && k==0) return 0;
        Init();
        for(int i=0;i<m;i++) scanf("%s",s),Insert(s,i);
        Build();
        memset(dp,0,sizeof(dp));
        dp[0][0][0]=1;
        int res = 0;
        for(int i=0;i<n;i++){
            for(int j=0;j<=tot;j++){
                for(int p=0;p<(1<<m);p++){
                    if(!dp[i][j][p]) continue;
                    for(int l=0;l<26;l++){
                        int v = nxt[j][l]; int st = sta[v];
                        add(dp[i+1][v][p|st],dp[i][j][p]);
                    }
                }
            }
        }
        for(int i=0;i<=tot;i++){
            for(int j=0;j<(1<<m);j++){
                if(cnt[j]>=k) add(res,dp[n][i][j]);
            }
        }
        res = (res+MOD)%MOD;
        printf("%d\n",res);
    }
    return 0;
}

你可能感兴趣的:(AC自动机,状压dp)