LA 3942 背单词 trie树+dp

题目：

题意：

给定一个字符串和给定一个单词集合。问从给定单词集合中选取单词，有多少种选取方法刚好拼接成字符串。

例如:

abcd

4

a

b

cd

ab

有两种

a-b-cd

ab-cd

分析：

刘汝佳书上入门经典的题目，看了他的模板敲了下，不错，我原先是用链表来存储的，以后就按这种形式了。

对于这题,要求的是有多少种方法，一看就是dp，其中d(i)表示从字符i开始的字符串的方案数S【i.....L】。则d(i)=sum{ d(i+len(x))|单词x是S【i....L】的前缀。

如何求单词x呢？当然是用trie数去匹配，将遍历到的单词节点找出来，然后do就可以了。

#include<iostream>
#include<cstring>
#include<vector>
#include<cstdio>
using namespace std;
const int maxnode=400000+10;
struct Trie
{
int ch[maxnode][26];
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;
}
void search(const char *s,int len,vector<int>& 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])ans.push_back(val[u]);
}
}
};
const int mod=20071027;
const int maxw=4000+10;
const int maxl=300000+10;
int f[maxl],len[maxw];
char text[maxl],word[110];
Trie trie;
int main()
{
int S,cas=0;
while(scanf("%s%d",text,&S)==2){
trie.clear();
for(int i=1;i<=S;i++){
scanf("%s",word);
len[i]=strlen(word);
trie.insert(word,i);
}
memset(f,0,sizeof(f));
int n=strlen(text);
f
=1;
for(int i=n-1;i>=0;i--){
vector<int>p;
trie.search(text+i,n-i,p);
for(int j=0;j<p.size();j++)
f[i]=(f[i]+f[i+len[p[j]]])%mod;
}
printf("Case %d: %d\n",++cas,f[0]);
}
return 0;
}