题目描述
输入描述
输出描述
输入样例
5
a
b
c
ad
acd
3
a
a
e
输出样例
OK
REPEAT
WRONG
题目大意:对输入的 n 个名字建 Trie 树,并在 m 个询问内查找是否有对应的 Trie 树,同时注意是否为第一次询问该名字。
快乐模板题,唯一在建树和查找之上增加的内容是判断当某个 Trie 树存在时,标记它是否已经访问过。可用 1 表示第一次找到,2 表示已找到过一次即可。
参考代码
#include <bits/stdc++.h>
using namespace std;
const int N = 5e5+10;
char s[60];
int ch[N][26],st[N];
int cnt = 1;
int main() {
int n,m;
cin>>n;
for(int i=1; i<=n;++i){
scanf("%s",s+1);
int p=1;
for(int j=1;s[j];++j){
int c=s[j]-'a';
if(!ch[p][c])
ch[p][c]=++cnt;
p=ch[p][c];
}
st[p] = 1;
}
cin>>m;
while (m--) {
scanf("%s", s + 1);
int p = 1;
for(int j=1;s[j];++j) {
int c=s[j]-'a';
p=ch[p][c];
if(!p)
break;
}
if(st[p]==1){
st[p]=2;
cout<<"OK"<<endl;
}
else if(st[p]==2)
cout<<"REPEAT"<<endl;
else
cout<<"WRONG"<<endl;
}
return 0;
}