题意:
首先要写一个添加字符串操作,比如说我输入“1 abc”就是添加字符串abc。然后写查询城市名称操作比如说我输入“2 abcabc”就是查询abcabc是否是所有添加过的字符串中前后缀组合(或本身就是某单词的前缀或后缀),输出“YES”或“NO”。
字符串不会超过100个字符!操作总数不超过50000个。
解析:
利用Trie构建两个字典树,前缀树和后缀树,先将字符串正着和插入前缀树中,再将字符串翻转插入后缀树中。然后每次查找每个字符前缀和后缀最长能抵达的位置,分别记为len1,len2,记当前字符串的长度为len,如果len1+len2>=len,则表面存在字符串的前缀和后缀能构成当前字符串。
my code
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxnode = 500005;
const int sigma_size = 26;
struct Trie {
int ch[maxnode][sigma_size];
int sz;
Trie() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}
void init() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}
int idx(char c) { return c - 'a'; }
void insert(char *s) {
int u = 0, n = strlen(s);
for(int i = 0; i < n; i++) {
int c = idx(s[i]);
if(!ch[u][c]) {
memset(ch[sz], 0, sizeof(ch[sz]));
ch[u][c] = sz++;
}
u = ch[u][c];
}
}
int find(char *s) {
int u = 0, n = strlen(s);
int ret = 0;
for(int i = 0; i < n; i++) {
int c = idx(s[i]);
if(!ch[u][c])
return ret;
u = ch[u][c];
ret++;
}
return ret;
}
} pre, suf;
int oper;
char str[105];
int n;
int main() {
while(scanf("%d", &n) != EOF) {
pre.init(), suf.init();
int len, len1, len2;
while(n--) {
scanf("%d%s", &oper, str);
len = strlen(str);
if(oper == 1) {
pre.insert(str);
reverse(str, str+len);
suf.insert(str);
}else {
len1 = pre.find(str);
reverse(str, str+len);
len2 = suf.find(str);
printf("%s\n", (len1 + len2 >= len) ? "YES" : "NO");
}
}
}
return 0;
}