【LA3942-Remember the word 】Trie

http://acm.hust.edu.cn/vjudge/problem/22109

题意：给定n个单词，一个字符串，问把这个字符串划分为若干个单词的连接（单词可重复使用）有多少种方案（mod20071027）。

题解：

设d[i]为从第i个字母开始的后缀有多少种匹配方案。
if 单词x是当前后缀的前缀: d[i]=sum{d(i+len(x))}

用所有单词建一棵trie，从后往前循环i，对于每一个i进入trie查找前缀，更新d[i]。

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

const int L=300100,Mod=20071027;
char s0[L],s[L];
int n,num;
int d[L],ch[L][30],bk[L];

void insert(char *c)
{
int l=strlen(c);
int x=0;
for(int i=0;i<l;i++)
{
int id=c[i]-'a'+1;
if(!ch[x][id])
{
num++;
ch[x][id]=num;
}
x=ch[x][id];
}
bk[x]=l;
}

void find(int l,int r)
{
int x=0;
for(int i=l;i<=r;i++)
{
int id=s0[i]-'a'+1;
if(!ch[x][id]) break;
x=ch[x][id];
if(bk[x]) d[l]=(d[l]+d[l+bk[x]])%Mod;
}
}

//d[i]=sum(d[i]+len(x))
int main()
{
freopen("a.in","r",stdin);
freopen("a.out","w",stdout);
int T=0;
while(scanf("%s",s0)!=EOF)
{
num=0;
memset(bk,0,sizeof(bk));
memset(ch,0,sizeof(ch));
memset(d,0,sizeof(d));
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",s);
insert(s);
}
int l=strlen(s0);
d[l]=1;
for(int i=l-1;i>=0;i--) find(i,l-1);
// for(int i=0;i<l;i++) printf("%d ",d[i]);printf("\n");
printf("Case %d: %d\n",++T,d[0]%Mod);
}

return 0;
}