题意:
给定一个长度为N≤104的加密句子,加密方式为将句子每个单词翻转,并移除空格,句子字母全为小写
现给定M≤105个单词wi,大小写均有,|wi|≤103且∑wi≤106
现用这些单词解密句子,每个单词可以用多次,保证一种解密方式存在,多解输出任意一种
分析:
一上来把单词数看错,看成1W个,写了个kmp预处理转移过程优化n3 dp到n2,终测T了
讨论题目发现是10W个单词,搞了个ac自动机上去,T更后面的case,−−,爆炸,果然记录转移是不行的
仔细想想这个dp,f[i]:=以i结尾的字符串是否可以解密,为记录答案直接存单词长度就可以了
其实直接在trie上顺推转移就可以了,没必要记录转移,只要1个答案本来就是n2的dp,被我强行搞成n3
时间复杂度为O(n∗|wmax|)
其实判断这个单词在母串中是否出现过完全没必要用trie啊,直接hash单词,维护母串前缀哈希,然后顺推转移就可以了
trie 代码:
//
// Created by TaoSama on 2016-02-27
// Copyright (c) 2016 TaoSama. All rights reserved.
//
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>
using namespace std;
#define pr(x) cout << #x << " = " << x << " "
#define prln(x) cout << #x << " = " << x << endl
const int N = 1e5 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;
const int M = 1e6 + 10, S = 26;
int n, m;
string s, t[N];
int f[N];
struct AcAutomata {
int root, sz;
int nxt[M][S], id[M];
int newNode() {
id[sz] = -1;
memset(nxt[sz], -1, sizeof nxt[sz]);
return sz++;
}
void init() {
sz = 0;
root = newNode();
}
void insert(string & s, int x) {
int u = root;
for(int i = 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 = root;
for(int i = idx; i < s.size(); ++i) {
u = nxt[u][s[i] - 'a'];
if(u == -1) return;
if(~id[u] && !f[i + 1])
f[i + 1] = id[u];
}
}
} ac;
int main() {
#ifdef LOCAL
freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin);
// freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout);
#endif
ios_base::sync_with_stdio(0);
cin >> n >> s >> m;
ac.init();
for(int i = 1; i <= m; ++i) {
cin >> t[i];
ac.insert(t[i], i);
}
f[0] = 1;
for(int i = 0; i < n; ++i) {
if(!f[i]) continue;
ac.query(i);
}
vector<int> ans;
int u = n;
while(u) {
ans.push_back(f[u]);
u -= t[f[u]].size();
}
for(int i = ans.size() - 1; ~i; --i)
cout << t[ans[i]] << (" \n"[!i]);
return 0;
}
sb数错0了写的kmp代码,就算不数错也得T
10W个a的单词,直接炸裂map记录的预处理,明明可以继续利用题目信息的,too young
TLE on test 67 代码:
//
// Created by TaoSama on 2016-02-27
// Copyright (c) 2016 TaoSama. All rights reserved.
//
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>
using namespace std;
#define pr(x) cout << #x << " = " << x << " "
#define prln(x) cout << #x << " = " << x << endl
const int N = 1e5 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;
int n, m, nxt[N];
string s, t[N];
map<pair<int, int>, int> mp;
void getNxt(string& t) {
nxt[0] = -1;
int i = 0, j = -1;
while(i < t.size()) {
if(j == -1 || t[i] == t[j]) nxt[++i] = ++j;
else j = nxt[j];
}
}
void kmp(string t, int id) {
reverse(t.begin(), t.end());
for(int i = 0; i < t.size(); ++i) t[i] = tolower(t[i]);
int i = 0, j = 0, sz = t.size();
getNxt(t);
while(i < s.size()) {
if(j == -1 || s[i] == t[j]) ++i, ++j;
else j = nxt[j];
if(j == sz) mp[ {i - sz + 1, i}] = id;
}
}
int f[N];
pair<int, int> pre[N];
int main() {
#ifdef LOCAL
freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin);
// freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout);
#endif
ios_base::sync_with_stdio(0);
cin >> n >> s >> m;
for(int i = 1; i <= m; ++i) {
cin >> t[i];
kmp(t[i], i);
}
f[0] = 1;
for(int i = 1; i <= n; ++i) {
for(int j = 0; j < i; ++j) {
if(!f[j]) continue;
if(mp.count({j + 1, i})) {
f[i] = 1;
pre[i] = {j, mp[{j + 1, i}]};
break;
}
}
}
vector<int> ans;
int u = n;
while(u) {
ans.push_back(pre[u].second);
u = pre[u].first;
}
for(int i = ans.size() - 1; ~i; --i)
cout << t[ans[i]] << (" \n"[!i]);
return 0;
}