Remember the Word UVALive - 3942 字典树 + 动态规划(Trie + dp)

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

题意：

给出一个由小写字母组成的长串和若干个小串，求由这些小串来组成这个长串共有多少种可能。

思路：

计数问题很容易想到 dp，但本题使用纯 dp 会超时，原因在于每次小串和长串匹配太耗时。我们可以使用字典树来保存这些小串，使得我们匹配时更新快速。设长串用$s$表示，设用数组 $\text{res}[MAXN]$ 来保存中间结果，$\text res[i]$ 表示$s[0\ldots\text{i-1}]$可能的组合方式种数。则状态转移方程为：$\text{res}[i] = sum(\text{res}[i - len(t)]),t = s[i - len(t)\ldots i]$。最终$res[len(s)]$即为所求答案。

代码：

//纯 dp 会 TLE
#include <bits/stdc++.h>
#define fp(PP,QQ,RR) for(int PP = QQ;PP < RR;PP ++)
#define MAXL 300005
#define MAXN 4005
using namespace std;
typedef long long ll;
ll res[MAXL],ro,n;
string s,wds[MAXN];
void solve(){
    res[0] = 1;
    cin >> n;
    fp(i,0,n){
        cin >> wds[i];
    }
    fp(i,0,s.size()){
        fp(j,0,n){
            if(s.substr(i,wds[j].size()) == wds[j]){
                res[i + wds[j].size()] = (res[i] + res[i + wds[j].size()]) % 20071027;
            }
        }
    }
}
int main(void)
{
    while(cin >> s){
        solve();
        printf("Case %lld: %lld\n",++ ro,res[s.size()]);
    }
    return 0;
}

//Trie + dp 顺利 AC
#include <bits/stdc++.h>
#define fp(PP,QQ,RR) for(int PP = QQ;PP < RR;PP ++)
#define MAXL 300005
#define MAXN 4005
using namespace std;
typedef long long ll;
int ro,n;
int id(char c){return c - 'a';}
struct trie{
    int ch[MAXN * 100][26];
    string s;
    ll res[MAXL],val[MAXN * 100],sz;
    void init(){
        memset(res,0,sizeof res);
        memset(val,0,sizeof val);
        memset(ch,0,sizeof ch);
        sz = 1;
        res[0] = 1;
    }
    void insert(string wd){
        int u = 0,len = wd.size();
        fp(i,0,len){
            int c = id(wd[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] ++;
    }
    ll query(){
        fp(i,0,s.size()){
            int u = 0,cur = i;
            for(int j = 1;ch[u][id(s[cur])] && cur < s.size();j ++,cur ++){
                u = ch[u][id(s[cur])];
                if(val[u])
                    res[cur + 1] = (res[cur + 1] + res[cur + 1 - j] * val[u]) % 20071027;
            }
        }
        return res[s.size()];
    }
}t;
int main(void)
{
    std::ios::sync_with_stdio(false);
    while(cin >> t.s){
        t.init();
        cin >> n;
        fp(i,0,n){
            string wd;
            cin >> wd;
            t.insert(wd);
        }
        printf("Case %d: %lld\n",++ ro,t.query());
    }
    return 0;
}