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[n]=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;
}