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

来源：互联网发布：nodejs access数据库编辑：程序博客网时间：2024/05/16 09:01

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

思路：不难想到递推方程 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;}

0 0