LA 3942 背单词（前缀树Trie+dp）

给出一个由S个不同的单词组成的字典和一个长字符串。把这个字符串分解成若干个单词的连接（单词可以重复使用），有多少种方法？比如：有4个单词a,b,cd,ab,则abcd有两种分解方法：a+b+cd和ab+cd。

输入格式：
多组数据，每组数据第一行为小写字母组成的待分解字符串，长度L不超过300000.第二行为单词个数S（1<=s<=4000）。以下S行分别为一个单词，由不超过100个小写字母组成。输入结束EOF。

分析：不难想到递推计算方法，dp[n]=sum(dp[i+len[j]]),单词j表示S[i…L]的前缀。
如果先枚举j，在判断是否为s[i…L]的前缀，则时间O(4000*300000)超时，因此换个思路，先把所有单词组成一棵Trie数，有关Trie数可以再下面博客查看。
http://www.cnblogs.com/huangxincheng/archive/2012/11/25/2788268.html
然后再在Trie数里面查前缀序列S[i…L]，查找没经过一个单词节点，就找到了一个上述状态的j值，时间O(100*300000)足够

代码首先用蓝书的Trie树模板，

struct Trie
{
    int ch[maxnode][sig_size];  //保存节点i的那些编号为j的子节点
    int val[maxnode];  //节点的附加信息
    int sz;  //节点总数
    void clear()
    {
        sz=1;    //初始时只有一个根节点
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c)
    {
        return c-'a';    //字符c的编号
    }
    //插入字符串s，附加信息为v，注意v必须非0，因为0代表本节点不是单词节点
    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]);
        }
    }
};

然后将数据进行树插入，形成Trie树，再查找每个单词节点的权值，代入转移方程中

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <iostream>
using namespace std;
#define maxnode 400010
const int sig_size=101;
struct Trie
{
    int ch[maxnode][sig_size];  //保存节点i的那些编号为j的子节点
    int val[maxnode];  //节点的附加信息
    int sz;  //节点总数
    void clear()
    {
        sz=1;    //初始时只有一个根节点
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c)
    {
        return c-'a';    //字符c的编号
    }
    //插入字符串s，附加信息为v，注意v必须非0，因为0代表本节点不是单词节点
    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 maxs =4010;
const int maxl = 300010;
int dp[maxl],len[maxs];
char str[maxl],word[110];
Trie trie;
int main()
{
    int s;
    while(~scanf("%s%d",str,&s))
    {
        trie.clear();
        for(int i=1; i<=s; i++)
        {
            scanf("%s",word);
            len[i]=strlen(word);
            trie.insert(word,i);
        }
        memset(dp,0,sizeof(dp));
        int len1=strlen(str);
        dp[len1]=1;
        for(int i=len1-1; i>=0; i--)
        {
            vector<int>q;
            trie.search(str+i,len1-i,q);
            for(int j=0; j<q.size(); j++)
            {
                dp[i]=(dp[i]+dp[i+len[q[j]]])%mod;
            }
        }
        cout<<dp[0]<<endl;
    }
    return 0;
}
/*
abcd
4
a
b
ab
cd
*/