UVa 1401 Remember the Word(Trie+DP)

题意：输入一个字符串，再输入n个子串，计算出字符串被不同子串组成的方式有几种。

Trie字典树+DP

insert（）构造字典树，如果是子串，则val记录为1，否则为0，ch为字符的编号。

query（）计算字符串中每一个字符可能被子串组成的方式有几种，用sum[]记录。

#include <iostream>
#include <cstdio>
#include <cstring>
#define N 300010
#define M 4000010
#define mod 20071027
using namespace std;

int n, sz;
char s[N], str[110];
int ch[M][26], val[M], sum[N];

void init(){
    memset(ch[0], 0, sizeof(ch));
    sz = 1;
    int len = strlen(s);
    for(int i = 1; i <= len; ++ i) sum[i] = 0;
    sum[0] = 1;
}

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

void insert(){
    int u = 0, len = strlen(str);
    for(int i = 0; i < len; ++ i){
        int c = idx(str[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] = 1;
}

void query(){
    int len = strlen(s);
    for(int i = 1; i < len; ++i){
        int u = 0;
        for(int j = i; j < len; ++ j){
            int c = idx(s[j]);
            if(!ch[u][c]) break;
            u = ch[u][c];
            if(val[u])
                sum[j] = (sum[i - 1] + sum[j]) % mod;
        }
    }
}

int main() {
    int t = 0;
    while(scanf("%s", s + 1) != EOF){
        s[0] = '1';
        init();
        scanf("%d", &n);
        for(int i = 0; i < n; ++ i){
            scanf("%s", str);
            insert();
        }
        query();
        int len = strlen(s);
        printf("Case %d: %d\n", ++ t, sum[len - 1] % mod);
    }
    return 0;
}