BZOJ 1030 TRIE图+DP

貌似是论问题？、、

感觉挺好玩了，写起来也简单。。

https://files.cnblogs.com/proverbs/Trie%E5%9B%BE%E7%9A%84%E6%9E%84%E5%BB%BA_%E6%B4%BB%E7%94%A8%E4%B8%8E%E6%94%B9%E8%BF%9B.pdf

 

View Code 
 1 #include <cstring>
 2 #include <cstdio>
 3 #include <cstdlib>
 4 #include <iostream>
 5 #include <algorithm>
 6 
 7 #define N 111
 8 #define MOD 10007
 9 
10 using namespace std;
11 
12 struct TR
13 {
14     int son[26]; bool fg; int f;
15 }tr[N*N];
16 
17 int n,cnt,m,ans;
18 int q[N*N];
19 char str[N];
20 int dp[N][N*N][2];
21 
22 inline void insert(char *s)
23 {
24     int len=strlen(s+1);
25     int now=1;
26     for(int i=1;i<=len;i++)
27     {
28         if(!tr[now].son[s[i]-'A']) tr[now].son[s[i]-'A']=++cnt;
29         now=tr[now].son[s[i]-'A'];
30     }
31     tr[now].fg=true;
32 }
33 
34 inline void build()
35 {
36     int h=1,t=1,sta; q[1]=1;
37     while(h<=t)
38     {
39         sta=q[h++];
40         for(int i=0;i<26;i++)
41         {
42             int now=tr[sta].son[i],tmp;
43             if(sta==1) tmp=1;
44             else tmp=tr[tr[sta].f].son[i];
45             if(now==0) tr[sta].son[i]=tmp;
46             else
47             {
48                 tr[now].f=tmp;tr[now].fg|=tr[tmp].fg;
49                 q[++t]=now;
50             }
51         }
52     }
53 }
54 
55 inline void read()
56 {
57     cnt=1; tr[1].f=1;
58     scanf("%d%d",&n,&m);
59     for(int i=1;i<=n;i++)
60     {
61         scanf("%s",str+1);
62         insert(str);
63     }
64     build();
65 }
66 
67 inline void go()
68 {
69     dp[0][1][0]=1;
70     for(int i=0;i<m;i++)
71         for(int j=1;j<=cnt;j++)
72         {
73             if(dp[i][j][0])
74             {
75                 for(int k=0;k<26;k++)
76                 {
77                     if(tr[tr[j].son[k]].fg) dp[i+1][tr[j].son[k]][1]=(dp[i+1][tr[j].son[k]][1]+dp[i][j][0])%MOD;
78                     else dp[i+1][tr[j].son[k]][0]=(dp[i+1][tr[j].son[k]][0]+dp[i][j][0])%MOD;
79                 }
80             }
81             if(dp[i][j][1])
82             {
83                 for(int k=0;k<26;k++)
84                     dp[i+1][tr[j].son[k]][1]=(dp[i+1][tr[j].son[k]][1]+dp[i][j][1])%MOD;
85             }
86         }
87     for(int i=1;i<=cnt;i++) ans=(ans+dp[m][i][1])%MOD;
88     printf("%d\n",ans);
89 }
90 
91 int main()
92 {
93     read(),go();
94     return 0;
95 }

 

 

 

posted @ 2013-03-05 21:15  proverbs  阅读(698)  评论(0)  编辑  收藏  举报