UVa 1401 - Remember the Word（Trie + DP）

题意

给一个字符串，求几种方式组成，由给定的字典。

思路

这题纠结了好几天，一直想不懂是怎么递推的。

在今天早上的电路课上，终于灵光一闪。。。。

dp[i] 表示从第i个开始的字符串， dp[i]=sum(i+len(x)) ，x是这个字符串的前缀。

用以前的版本的字典树竟然无限TLE！姿势都一样。。看来以后只能用lrj版的字典树了。

代码

  
  
  
  

   
   
   
   
#include <cstdio>
   
   
   
   
#include <stack>
   
   
   
   
#include <set>
   
   
   
   
#include <iostream>
   
   
   
   
#include <string>
   
   
   
   
#include <vector>
   
   
   
   
#include <queue>
   
   
   
   
#include <functional>
   
   
   
   
#include <cstring>
   
   
   
   
#include <algorithm>
   
   
   
   
#include <cctype>
   
   
   
   
#include <ctime>
   
   
   
   
#include <cstdlib>
   
   
   
   
#include <fstream>
   
   
   
   
#include <string>
   
   
   
   
#include <sstream>
   
   
   
   
#include <map>
   
   
   
   
#include <cmath>
   
   
   
   
#define LL long long
   
   
   
   
#define lowbit(x) ((x) & (-x))
   
   
   
   
#define MP(a, b) make_pair(a, b)
   
   
   
   
#define MS(arr, num) memset(arr, num, sizeof(arr))
   
   
   
   
#define PB push_back
   
   
   
   
#define F first
   
   
   
   
#define S second
   
   
   
   
#define ROP freopen("input.txt", "r", stdin);
   
   
   
   
const double PI = acos(-1.0);
   
   
   
   
const int INF = 0x3f3f3f3f;
   
   
   
   
using namespace std;
   
   
   
   
const int MAXN = 400000 + 100;
   
   
   
   
const int MOD = 20071027;
   
   
   
   

   
   
   
   
typedef pair<int, int> pii;
   
   
   
   
typedef vector<int>::iterator viti;
   
   
   
   
typedef vector<pii>::iterator vitii;
   
   
   
   

   
   
   
   
int dp[300000 + 10], len;
   
   
   
   

   
   
   
   
struct TRIE
   
   
   
   
{
   
   
   
   
 int ch[MAXN][26];
   
   
   
   
 int v[MAXN];
   
   
   
   
 int sz;
   
   
   
   
 void init()
   
   
   
   
 {
   
   
   
   
 sz = 1;
   
   
   
   
 MS(ch[0], 0);
   
   
   
   
 }
   
   
   
   
 void insert(char *s)
   
   
   
   
 {
   
   
   
   
 int cur = 0, leng = strlen(s);
   
   
   
   
 for (int i = 0; i < leng; i++)
   
   
   
   
 {
   
   
   
   
 int idx = s[i] - 'a';
   
   
   
   
 if (!ch[cur][idx])
   
   
   
   
 {
   
   
   
   
 MS(ch[sz], 0);
   
   
   
   
 v[sz] = 0;
   
   
   
   
 ch[cur][idx] = sz++;
   
   
   
   
 }
   
   
   
   
 cur = ch[cur][idx];
   
   
   
   
 }
   
   
   
   
 v[cur] = 1;
   
   
   
   
 }
   
   
   
   
 void search(char *s, int x)
   
   
   
   
 {
   
   
   
   
 int cur = 0;
   
   
   
   
 for (int i = x; i < len; i++)
   
   
   
   
 {
   
   
   
   
 int idx = s[i] - 'a';
   
   
   
   
 if (!ch[cur][idx]) return;
   
   
   
   
 cur = ch[cur][idx];
   
   
   
   
 if (v[cur]) dp[x] = (dp[x] + dp[i + 1]) % MOD;
   
   
   
   
 }
   
   
   
   
 }
   
   
   
   
}trie;
   
   
   
   

   
   
   
   
char dic[110], str[300000 + 10];
   
   
   
   

   
   
   
   
int main()
   
   
   
   
{
   
   
   
   
 //ROP;
   
   
   
   
 int n, i, j, cases = 0;
   
   
   
   
 while (~scanf("%s", str))
   
   
   
   
 {
   
   
   
   
 len = strlen(str);
   
   
   
   
 scanf("%d", &n);
   
   
   
   
 trie.init();
   
   
   
   
 for (i = 0; i < n; i++)
   
   
   
   
 {
   
   
   
   
 scanf("%s", dic);
   
   
   
   
 trie.insert(dic);
   
   
   
   
 }
   
   
   
   
 dp[len] = 1;
   
   
   
   
 for (i = len - 1; i >= 0; i--)
   
   
   
   
 {
   
   
   
   
 dp[i] = 0;
   
   
   
   
 trie.search(str, i);
   
   
   
   
 }
   
   
   
   
 printf("Case %d: %d\n", ++cases, dp[0]);
   
   
   
   
 }
   
   
   
   
 return 0;
   
   
   
   
}