【暑假】[实用数据结构]UVAlive 3942 Remember the Word

UVAlive 3942 Remember the Word

 

题目：

Remember the Word

 

Time Limit: 3000MS

Memory Limit: Unknown

64bit IO Format: %lld & %llu

 

 

Submit Status

Description

Neal is very curious about combinatorial problems, and now here comes a problem about words. Knowing that Ray has a photographic memory and this may not trouble him, Neal gives it to Jiejie.

Since Jiejie can't remember numbers clearly, he just uses sticks to help himself. Allowing for Jiejie's only 20071027 sticks, he can only record the remainders of the numbers divided by total amount of sticks.

The problem is as follows: a word needs to be divided into small pieces in such a way that each piece is from some given set of words. Given a word and the set of words, Jiejie should calculate the number of ways the given word can be divided, using the words in the set.

Input

The input file contains multiple test cases. For each test case: the first line contains the given word whose length is no more than 300 000.

The second line contains an integer S , 1S4000 .

Each of the following S lines contains one word from the set. Each word will be at most 100 characters long. There will be no two identical words and all letters in the words will be lowercase.

There is a blank line between consecutive test cases.

You should proceed to the end of file.

Output

For each test case, output the number, as described above, from the task description modulo 20071027.

Sample Input

abcd 
4 
a 
b 
cd 
ab

Sample Output

Case 1: 2



思路：

 将一个字符串按照所给单词分解，求解有多少种分解方案。DP + Trie加速。定义d[i]为后缀i的分解方案数，可以得到状态转移方程：
  d[i] += d[i+len[x]]
 其中x为后缀i前缀的标号,len为标号为x的单词长度。
 但如何快速确定后缀i的前缀呢？这里用到了Trie，在find(后缀i）的路径上有val标记的都为后缀i的前缀，只需要O(n)的时间可以构造出p。

代码：

#include<cstdio>
#include<cstring>
#include<vector>
#define FOR(a,b,c) for(int a=(b);a<(c);a++)
using namespace std;

const int maxnode = 4000 * 100 + 10;
const int sigma_size = 26;

// 字母表为全体小写字母的Trie
struct Trie {
  int ch[maxnode][sigma_size];
  int val[maxnode];
  int sz; 
  void clear() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); } 
  int idx(char c) { return c - 'a'; } 

  void insert(const char *s, int v) {
    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] = v; 
  }

  // 找字符串s的长度不超过len的前缀
  void find_prefixes(const char *s, int len, vector<int>& ans) { //构造ans 
    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) ans.push_back(val[u]); // 找到一个前缀
    }
  }
  //find_prefixes操作以vector形式返回满足要求的字串标号val 
  //find操作需要对查找操作进行长度限制 以节省时间 
};

const int maxl= 300000 + 10;
const int maxw= 4000+10;    //单词个数
const int maxwl=100 + 10;
const int MOD = 20071027;

char text[maxl];
int d[maxl],len[maxw];
Trie trie;

int main(){
int kase=0,n;
  while(scanf("%s%d",text,&n)==2){
      trie.clear();   //clear()
      char word[maxwl];
      FOR(i,1,n+1){   //标号为1..n 
          scanf("%s",word);
          len[i]=strlen(word);    //记录word长度 节约了一个char[][] 
          trie.insert(word,i);   //构造Trie val为word标号 
      }

      // DP 
      memset(d,0,sizeof(d));
      int L=strlen(text);
      d[L]=1;   // ? :DP边界 只要后缀中有前缀单词 则 += 1 
      for(int i=L-1;i>=0;i--){
          vector<int> p;                     //存储 
          trie.find_prefixes(text+i,L-i,p); 
        //寻找在后缀text+i中以前缀形势出现过的所有单词 //text+i为字串text[i..L) 
          FOR(j,0,p.size())
            d[i] = (d[i]+d[i+len[p[j]]]) % MOD;  //i+len-1为单词尾 
      }
      printf("Case %d: %d\n",++kase,d[0]);
  }
  return 0;
}