UVALive 394２ Remember the Word 字典树DP&&前缀树

来源：互联网发布：淘宝返利红包链接编辑：程序博客网时间：2024/05/23 19:12

题目描述：给你一个由S个不同单词组成的字典和一个长字符串。把这个字符串分解成若干个单词连接（）单词可以重复使用），求有多少种方法。

Sol: 递推

dp[i]=sigma_sum (dp[i+len(x)])

dp[i]表示从字符i开始的字符串即后缀(s[i..L])的分解方案数。

x为是 (s[i..L]的前缀)。

先把所有单词放在trie上，然后在trie上查找S[i....L],没经过一个节点就找到了一个X。。。然后就可以解了。

#include <iostream>#include <cstring>#include <cstdio>using namespace std;const int MaxNode = 4000 * 100 + 10;const int sigma_size = 26;const int maxL = 300000 + 10;const int maxWL = 100 + 10;const int MOD = 20071027;int dp[maxL],N;char text[maxL],word[maxWL];struct Trie{int ch[MaxNode][sigma_size];int val[MaxNode];int sz;//结点总数 void clear() {sz=1;val[0]=0;memset(ch[0],0,sizeof(ch[0]));}int idx(char c) {return c-'a';}void insert(const char *s){int u=0,n=strlen(s);for(int i=0;i<n;i++){int c=idx(s[i]);if(!ch[u][c]){memset(ch[sz],0,sizeof(ch[sz]));val[sz]=0;ch[u][c]=sz++;}u=ch[u][c];}val[u]=n;}void find(const char *s,int len,int q){int u=0;for(int i=0;i<len;i++){if(s[i]=='\0')  break;int c=idx(s[i]);if(!ch[u][c])  break;u=ch[u][c];if(val[u]!=0)  dp[q]=(dp[q]+dp[q+val[u]])%MOD;}}};Trie trie;int main(){int Case=0;while(~scanf("%s%d",text,&N)){trie.clear();for(int i=0;i<N;i++){scanf("%s",word);trie.insert(word);}memset(dp,0,sizeof(dp));int L=strlen(text);dp[L]=1;for(int i=L-1;i>=0;i--){trie.find(text+i,L-i,i);}printf("Case %d: %d\n",++Case,dp[0]);}return 0;}