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

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

题意:

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

思路:

计数问题很容易想到 dp,但本题使用纯 dp 会超时,原因在于每次小串和长串匹配太耗时。我们可以使用字典树来保存这些小串,使得我们匹配时更新快速。设长串用 s s 表示,设用数组 res[MAXN] res [ M A X N ] 来保存中间结果, res[i] r e s [ i ] 表示 s[0i-1] s [ 0 … i-1 ] 可能的组合方式种数。则状态转移方程为: res[i]=sum(res[ilen(t)]),t=s[ilen(t)i] res [ i ] = s u m ( res [ i − l e n ( t ) ] ) , t = s [ i − l e n ( t ) … i ] 。最终 res[len(s)] r e s [ l e n ( 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;
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/qq_36501006/article/details/81808909
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞