LA3942 Remember the Word

题目链接：https://vjudge.net/problem/UVALive-3942

本篇是刘汝佳《算法竞赛入门经典——训练指南》的读书笔记（复述），详见原书 (P209) .

解题思路：

　　先用字典树维护字典中所有的单词。

　　定义 (f(x)) 为以长字符串中第 (x) 个字符开始的字符串的分解方案数，则 (f(x) = sum{f(x+len(i))}) ，其中 (i) 是该字符串的所有前缀。

　　用一种类似 (DFS) 的方式解决这个问题。

　　注意：书中的模数写错了。

AC代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const ll mod = 20071027;
const int maxn = 300000+5;
char inp[maxn];
struct Trie{
    int ch[400010][30];
    int val[400010];
    int sz;
    void inserts(char *s,int v){
        int u=0,n=strlen(s);
        for(int i=0;i<n;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;
    }
};
Trie tree;
ll f[maxn];

ll solve(int head,int tail){
    if(f[head]!=-1) return f[head];
    int now=0;
    ll ret=0;
    for(int i=head;i<tail;i++){
        int c=inp[i]-'a';
        int Next=tree.ch[now][c];
        if(Next==0){
            f[head]=ret;
            return ret;
        }
        if(tree.val[Next])
            ret=(ret+solve(i+1,tail))%mod;
        now=Next;
    }
    f[head]=ret;
    return ret;
}
int main(){
    char word[110];
    int kase=1;
    while(scanf("%s",inp)==1){
        int S;
        memset(f,-1,sizeof(f));
        tree.sz=1;
        memset(tree.ch[0],0,sizeof(tree.ch[0]));
        scanf("%d",&S);
        while(S--){
            scanf("%s",word);
            tree.inserts(word,1);
        }
        int tail=strlen(inp);
        f[tail]=1;
        printf("Case %d: %lld
",kase++,solve(0,tail));
    }
    return 0;
}