UVa1401 Remember the Word（DP+Trie树）

题目给定一个字符串集合有几种方式拼成一个字符串。

• dp[i]表示str_i...str_len-1的方案数
• dp[len]=1
• dp[i]=∑dp[j]（str_i...str_j-1∈SET）

用集合的字符串构造一棵Trie树，然后就可以在线性时间判断哪几个前缀字符串在集合里。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 int ch[][],tn;
 bool flag[];
 void insert(char *s){
     int x=;
     for(int i=; s[i]; ++i){
         int y=s[i]-'a';
         if(ch[x][y]==) ch[x][y]=++tn;
         x=ch[x][y];
     }
     flag[x]=;
 }
 char word[];
 int d[];
 int main(){
     int t=,n;
     char str[];
     while(~scanf("%s",word)){
         tn=;
         memset(ch,,sizeof(ch));
         memset(flag,,sizeof(flag));
         scanf("%d",&n);
         while(n--){
             scanf("%s",str);
             insert(str);
         }
         memset(d,,sizeof(d));
         int len=strlen(word);
         d[len]=;
         for(int i=len-; i>=; --i){
             int x=;
             for(int j=i; j<len; ++j){
                 int y=word[j]-'a';
                 if(ch[x][y]==) break;
                 x=ch[x][y];
                 if(flag[x]) d[i]+=d[j+],d[i]%=;
             }
         }
         printf("Case %d: %d\n",++t,d[]);
     }
     return ;
 }