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

题意：有一个长为n（n<= 25） 的字符串，它至少由k个magic word 组成，现在给出m个magic word，求出这个字符串组成的可能种数。

构造AC自动机进行搜索，dp【i】【j】【k】表示字符串长度为i，匹配字典树上的第j个节点，并且已经匹配上k个magic word时的总数。

则转移方程为 （dp【i+1】【j的儿子】【k  |  j的儿子的状态】 += dp【i】【j】【k】） % mod；

需要注意的是，因为单词可以重复使用，所以单词结尾的fail 指向root 指向的各节点

 

#include <iostream>

#include <algorithm>

#include <cmath>

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <string>

#include <vector>

#include <set>

#include <queue>

#include <stack>

#include <climits>//形如INT_MAX一类的

#define MAX 100005

#define mod 20090717

#define INF 0x7FFFFFFF

#define REP(i,s,t) for(int i=(s);i<=(t);++i)

#define LL long long

#define mem(a,b) memset(a,b,sizeof(a))

#define mp(a,b) make_pair(a,b)

#define L(x) x << 1

#define R(x) x << 1 | 1

# define eps 1e-5

//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂

using namespace std;



struct Trie {

    int next[26] ;

    int fail ;

    int buff ;

    void init() {

        memset(next,0,sizeof(next)) ;

        fail = -1 ;

        buff = 0 ;

    }

} a[111];

int cnt,n,m,K;

int q[111] , dp[30][111][1 << 10];

char keyword[22];



void insert(char *s,int pos) {

    int p = 0 ;

    for(int i = 0 ; s[i] ; i ++) {

        int t = s[i] - 'a' ;

        if(a[p].next[t] == 0) {

            a[cnt].init();

            a[p].next[t] = cnt ++;

        }

        p = a[p].next[t] ;

    }

    a[p].buff = 1 << pos;

}

void ac_bfs() {

    int i,head = 0,tail = 0;

    q[tail ++] = 0;

    while(head < tail) {

        int front = q[head ++];

        for(i = 0; i < 26 ; i ++) {

            if(a[front].next[i] == 0) {

                if(front != 0) a[front].next[i] = a[a[front].fail].next[i];

            } else {

                int p = a[front].fail ;

                while(p != -1) {

                    if(a[p].next[i] != 0) {

                        a[a[front].next[i]].fail = a[p].next[i] ;

                        a[a[front].next[i]].buff |= a[a[p].next[i]].buff;//状态合并，因为如果fail指向的节点也是某个magic word的结尾，那么相当于一次找到两个magic word

                        break ;

                    }

                    p = a[p].fail ;

                }

                if(p == -1) a[a[front].next[i]].fail = 0 ;

                q[tail ++] = a[front].next[i] ;

            }

        }

    }

}



int calone(int x) {

    int cal = 0;

    while(x) {

        if(x & 1) cal ++;

        x = x >> 1;

    }

    return cal;

}



void solve() {

    dp[0][0][0] = 1;

    int total = 1 << m;

    for(int i=0; i<n; i++) {

        for(int j=0; j<cnt; j++) {

            for(int k=0; k<total; k++) {

                if(dp[i][j][k] == 0) continue;

                for(int l=0; l<26; l++) {

                    int buff = a[a[j].next[l]].buff;

                    dp[i+1][a[j].next[l]][k | buff] = (dp[i+1][a[j].next[l]][k | buff] + dp[i][j][k]) % mod;

                }

            }

        }

    }

    int ans = 0;

    for(int j=0; j<cnt; j++) {

        for(int k=0; k<total; k++) {

            if(calone(k) < K) continue;

            ans = (ans + dp[n][j][k] ) % mod;

        }

    }

    printf("%d\n",ans);

}



int main() {

    while(scanf("%d%d%d",&n,&m,&K)) {

        if(n == 0 && m == 0 && K == 0) break;

        a[0].init();

        cnt = 1;

        for(int i=0; i<m; i++) {

            scanf("%s",keyword);

            insert(keyword,i);

        }

        ac_bfs();

        for(int i=0; i<=n; i++) {

            for(int j=0; j<cnt; j++) {

                for(int l=0; l<(1<<m); l++) {

                    dp[i][j][l] = 0;

                }

            }

        }

        solve();

    }

    return 0;

}

 

现在缺少dp的思路和灵感.......照着别人的思路敲的.....有些地方依旧有待解决