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

系统 1552 0

题意:有一个长为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的思路和灵感.......照着别人的思路敲的.....有些地方依旧有待解决

 

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


更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描下面二维码支持博主2元、5元、10元、20元等您想捐的金额吧,狠狠点击下面给点支持吧,站长非常感激您!手机微信长按不能支付解决办法:请将微信支付二维码保存到相册,切换到微信,然后点击微信右上角扫一扫功能,选择支付二维码完成支付。

【本文对您有帮助就好】

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描上面二维码支持博主2元、5元、10元、自定义金额等您想捐的金额吧,站长会非常 感谢您的哦!!!

发表我的评论
最新评论 总共0条评论