UVA 1401 & LA 3942 Remember the Word 字典树+DP

题意：给你一个字符串 再给你若干单词 问你这些单词有多少种组合方式可以组合成所给的字符串

思路：不难想到递推方程 dp[i] = sum(dp[i+len(x)]|x是从i开始的字符串的前缀单词) 问题便转化为了求一个单词是否是一个字符串的前缀  如果按照朴素做法 对于每一个子串都要枚举每一个单词 时间复杂度为3×10^5×4000×100 必然要T 如果我们将单词存储到字典树里 对每个子串的查询只需要100 时间复杂度降为3×10^5×100 可以接受。。。

#include <cstdio>
#include <cstring>
using namespace std;

#define REP( i, a, b ) for( int i = a; i < b; i++ )
#define REV( i, a, b ) for( int i = a; i >= b; i-- )
#define CLR( a, x ) memset( a, x, sizeof a )

const int maxn = 300010;
const int MOD = 20071027;

struct Trie{
          bool is_end;
          Trie *next[26];
          Trie() : is_end(false) { REP(i, 0, 26) next[i] = NULL; }
};

int n, dp[maxn], cas = 0;
char str[maxn];

void CreatTrie(Trie *root, char *s){
          int len = strlen(s);
          Trie *p = root, *q;
          REP(i, 0, len){
                    int id = s[i] - 'a';
                    if(p -> next[id] == NULL){
                              q = new Trie;
                              p -> next[id] = q;
                              p = p -> next[id];
                    }
                    else p = p -> next[id];
          }
          p -> is_end = true;
}

void FindTrie(Trie *root, int cur, int len){
          Trie *p = root;
          REP(i, cur, len){ //每次以cur的位置从字典树的根部开始查找
                    int id = str[i] - 'a';
                    if(p -> next[id] == NULL) return;
                    if(p -> next[id] -> is_end) dp[cur] = (dp[cur] + dp[i + 1]) % MOD;
                    p = p -> next[id];
          }
}

void DelTrie(Trie *root){
          if(root == NULL) return;
          REP(i, 0, 26) DelTrie(root -> next[i]);
          delete[] root;
}

void solve(){
          int len = strlen(str);
          char word[110];
          Trie *root = new Trie;
          CLR( dp, 0 );
          scanf("%d", &n);
          REP(i, 0, n){
                    scanf("%s", word);
                    CreatTrie(root, word);
          }
          dp[len] = 1;
          REV(i, len - 1, 0) FindTrie(root, i, len);
          printf("Case %d: %d\n", ++cas, dp[0]);
          DelTrie(root);
}

int main()
{
          //freopen("in.txt", "r", stdin);
          while(~scanf("%s", str)) solve();
          return 0;
}