【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=,Mod=;
char s0[L],s[L];
int n,num;
int d[L],ch[L][],bk[L];

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

void find(int l,int r)
{
    int x=;
    for(int i=l;i<=r;i++)
    {
        int id=s0[i]-'a'+;
        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=;
    while(scanf("%s",s0)!=EOF)
    {
        num=;
        memset(bk,,sizeof(bk));
        memset(ch,,sizeof(ch));
        memset(d,,sizeof(d));
        scanf("%d",&n);
        for(int i=;i<=n;i++)
        {
            scanf("%s",s);
            insert(s);
        }
        int l=strlen(s0);
        d[l]=;
        for(int i=l-;i>=;i--) find(i,l-);
        // for(int i=0;i<l;i++) printf("%d ",d[i]);printf("\n");
        printf("Case %d: %d\n",++T,d[]%Mod);
    }

    return ;
}