Manthan, Codefest 16 C. Spy Syndrome 2 字典树 + dp

最新推荐文章于 2018-08-09 17:07:45 发布

原创最新推荐文章于 2018-08-09 17:07:45 发布 · 561 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

ACM/ICPC_Codeforences 同时被 3 个专栏收录

204 篇文章

订阅专栏

ACM/ICPC_动态规划

120 篇文章

订阅专栏

数据结构_字典树

5 篇文章

订阅专栏

本文介绍了解决CodeForces 633C问题的方法，该问题要求通过翻转并拼接一系列单词形成一个字符串，并找出原始单词。采用字典树与动态规划相结合的技术，首先反转输入字符串，再利用字典树查找匹配的单词段，通过动态规划确定可行的分割方案。

题目链接：http://codeforces.com/contest/633/problem/C
题意：
你有一堆单词，然后把这一堆单词都翻转了，然后拼成了一个串。

然后现在给你一个串，让你找到原来拼的那些单词是什么。
解法：
字典树+dp
直接把那个串翻转一下，就相当于倒着做了嘛
然后我们跑dp就好了，dp[i]表示这个位置能否被转移到
然后我们就开始在字典树上面跑啊跑，看最远能够跑到哪儿，跑到了都打一个dp[i]=1，输出不太好说，看代码很容易看懂
注意大小写……

//CF 633C

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
const int maxm = 1e6+10;
int n, m, dp[maxn];
string s, t[maxn];
struct Trie{
    int rt, sz, nxt[maxm][26], id[maxm];
    int newnode(){
        id[sz] = -1;
        memset(nxt[sz], -1, sizeof(nxt[sz]));
        return sz++;
    }
    void init(){
        sz = 0;
        rt = newnode();
    }
    void insert(string &s, int x){
        int u = rt;
        for(int i = (int)s.size() - 1; ~i; i--){
            int c = tolower(s[i]) - 'a';
            int &v = nxt[u][c];
            if(v == -1) v = newnode();
            u = v;
        }
        id[u] = x;
    }
    void query(int idx){
        int u = rt;
        for(int i = idx; i < (int)s.size(); i++){
            u = nxt[u][s[i]-'a'];
            if(u == -1) return;
            if(~id[u] && !dp[i+1]) dp[i+1] = id[u];
        }
    }
}trie;

int main()
{
    cin>>n>>s>>m;
    trie.init();
    for(int i=1; i<=m; i++){
        cin>>t[i];
        trie.insert(t[i],i);
    }
    dp[0]=1;
    for(int i=0; i<n; i++){
        if(!dp[i]) continue;
        trie.query(i);
    }
    vector<int>ans;
    int u=n;
    while(u){
        ans.push_back(dp[u]);
        u-=t[dp[u]].size();
    }
    for(int i=ans.size()-1; ~i; i--){
        cout<<t[ans[i]]<<" ";
    }
    cout<<endl;
}