LA 3942 背单词

https://vjudge.net/problem/UVALive-3942

题意：

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

思路：

建立字典树，查询的时候令d(i)表示从字符i开始的字符串的分解方案数，则d(i)=sum{d(i+len(x))|单词x是S[i..L]的前缀}。

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;

const int maxnode = 4000 * 100 + 10;
const int sigma_size = 26;
const int maxn = 300000 + 10;
char str[maxn],s[100+5];
int n, dp[maxn];


struct Trie
{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;

    void init()
    {
        sz = 1;
        memset(ch[0], 0, sizeof(ch[0]));
    }

    void insert(char *s, int v)
    {
        int u = 0, len = strlen(s);
        for (int i = 0; i < len; i++)
        {
            int c = s[i] - 'a';
            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(char *s, int pos)
    {
        int len = strlen(s), u = 0, ans = 0;
        for (int i = pos; i < len; i++)
        {
            int c = s[i] - 'a';
            if (ch[u][c] == 0)  break;
            u = ch[u][c];
            if (val[u] == 1)
                ans = (ans + dp[i + 1]) % 20071027;
        }
        return ans;
    }
}T;


int main()
{
    //freopen("D:\txt.txt", "r", stdin);
    int kase = 0;
    while (~scanf("%s", str))
    {
        scanf("%d", &n);
        T.init();
        for (int i = 0; i < n; i++)
        {
            scanf("%s", s);
            T.insert(s, 1);
        }
        int len = strlen(str);
        dp[len] = 1;
        for (int i = len - 1; i >= 0; i--)
            dp[i] = T.find(str, i);
        printf("Case %d: %d
", ++kase, dp[0]);
    }
    return 0;
}