UVALive 3942 Remember the Word（trie + dp）

题目大意：给你一个字符串str，然后给你n个单词，让你用这些单词去组成这个字符串，问你有几种分解方法？

思路：先开始是一个dp，设 d[ i ] 表示从 i~len-1 的方案数（注意是后缀），有关系式 d[ i ] = sum(d[ i + x])，其中这 x 个字符是一个单词。但是如果朴素的做，肯定TLE，有4000个单词，长度最长是100，str的长度也为300000,。所以这里用trie，这样，算 sum 的时候，一遍找下去就行了，复杂度为 100。

因为一直习惯于dp写前缀，这样的话，就不能用 trie 了，原来转化为 后缀就行了，然后再用 trie 来优化，长见识了。。。 = =

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int MAXN = 333333;

const int MOD = 20071027;

char str[MAXN];

char word[111];

const int MAX_NODE = 4000*101;
const int SIGMA_SIZE = 26;

int d[MAXN];

int ch[MAX_NODE][SIGMA_SIZE];

struct Trie
{

    int sz;
    int val[MAX_NODE];

    Trie()
    {
        val[0] = 0;
        sz = 1;
        memset(ch[0],0,sizeof(ch[0]));
    }

    int idx(char c) {return c - 'a';}

    void insert(char *s,int v)
    {
        int len = strlen(s);
        int u = 0;
        for(int i = 0;i<len;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;
    }

    int find(int x,int len)
    {
        int u = 0;
        int cnt = 0 ;
        for(int i = x;i<=len;i++)
        {
            if(val[u]) cnt = (cnt + d[i])%MOD;
            int c = idx(str[i]);
            if(!ch[u][c]) return cnt;
            u = ch[u][c];
        }
        return cnt;
    }
};

int main()
{
    int cas = 0;
    while(~scanf("%s",str))
    {
        Trie tr;
        int m;
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",word);
            tr.insert(word,1);
        }
        int len = strlen(str);
        d[len] = 1;
        for(int i = len-1;i>=0;i--)
        {
            d[i] = tr.find(i,len);
        }
        printf("Case %d: %d\n",++cas,d[0]);
    }
    return 0;
}