UVA 1401 Remember the Word

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=36217

问一个单词可以如何用所给的一系列单词拆分

问题分析：

a bcd

ab cd

abc d

abcd

很容易想到递归：dp[i] = sum(dp[i+len])其中dp[i]是指字符串从i开始的后缀有多少种拆分方式，len是字典中给出的前缀的长度，由此想到用前缀树来以O(100)的复杂度求出len即可，注意dp的初始化。

AC代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>

using namespace std;

const int LEN = 300010;
const int MOD = 20071027;
const int MAX = 4000*110;

char str[LEN];
int cnt,dp[LEN],ans[110],num,len;

struct Trie{
    bool Isword;
    int next[110];
}node[MAX];

void Init(){
    cnt = 1;
    for(int i=0; i<MAX; i++){
        memset(node[i].next, -1, sizeof(node[i].next));
        node[i].Isword = false;
    }
}

void Insert(char *s){
    int p = 0, n = strlen(s);
    for(int i=0; i<n; i++){
        if(node[p].next[s[i]-'a'] == -1)
            node[p].next[s[i]-'a'] = cnt++;
        p = node[p].next[s[i]-'a'];
    }
    node[p].Isword = true;
}

void Query(int pos){
    num = 0;///记录有几个可用前缀
    memset(ans, 0, sizeof(ans));
    int tmp = 0, p = node[0].next[str[pos]-'a'];
    for(int i=pos+1; i<=len; i++){
        tmp++;
        if(p == -1) break;
        if(node[p].Isword) ans[num++] = tmp;///记录每个可用前缀的长度
        if(i == len) break;
        p = node[p].next[str[i]-'a'];
    }
    return;
}

int main(){
    int t = 0;
    while(cin >> str){
        Init();
        int n; cin >> n;
        char tmp[110];
        for(int i=0; i<n; i++){
            scanf("%s",tmp);
            Insert(tmp);
        }
        memset(dp, 0, sizeof(dp));
        len = strlen(str), dp[len] = 1;
        for(int i=len-1; i>=0; i--){
            Query(i);
            for(int j=0; j<num; j++){
                if(ans[j]){
                    dp[i] += dp[i+ans[j]] % MOD;
                    dp[i] %= MOD;
                }
            }
        }
        printf("Case %d: ",++t);
        printf("%d\n", dp[0]);
    }
    //system("pause");
    return 0;
}