题目链接:P2580「于是他错误的点名开始了」 。
XS中学化学竞赛组教练是一个酷爱炉石的人。
他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次,然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉(详情请见已结束比赛 CON900)。
这之后校长任命你为特派探员,每天记录他的点名。校长会提供化学竞赛学生的人数和名单,而你需要告诉校长他有没有点错名。(为什么不直接不让他玩炉石。)
第一行一个整数 n,表示班上人数。
接下来 n行,每行一个字符串表示其名字(互不相同,且只含小写字母,长度不超过 50)。
第 n+2 行一个整数 m,表示教练报的名字个数。
接下来 m 行,每行一个字符串表示教练报的名字(只含小写字母,且长度不超过 50)。
对于每个教练报的名字,输出一行。
如果该名字正确且是第一次出现,输出 OK
,如果该名字错误,输出 WRONG
,如果该名字正确但不是第一次出现,输出 REPEAT
。
5
a
b
c
ad
acd
3
a
a
e
OK
REPEAT
WRONG
定睛一看,字典树板子题,结束。
#include <bits/stdc++.h>
using namespace std;
// 字典树
struct Trie {
#ifndef _TRIE_
#define ll int
#define MAXN 500005
#define MAXCHAR 26
#endif
ll cnt; // 动态开点(开 Trie 树结点编号)
ll next[MAXN][MAXCHAR]; // 记录关联边为对应字符的子结点的下表
ll exist[MAXN]; // 记录以该结点结尾的字符串是否存在
Trie(): cnt(0) {}
// 插入字符串
void insert(char *str, ll len) {
ll p = 0;
for(ll i = 0; i < len; ++i) {
ll c = str[i] - 'a';
if(!next[p][c]) {
next[p][c] = ++cnt;
}
p = next[p][c];
}
exist[p] = 1;
}
// 查找字符串
ll find(char *str, ll len) {
ll p = 0;
for(ll i = 0; i < len; ++i) {
ll c = str[i] - 'a';
if(!next[p][c]) {
return false;
}
p = next[p][c];
}
return exist[p] ? exist[p]++ : exist[p];
}
};
int main()
{
ll n, m;
char str[MAXN];
static Trie t;
scanf("%d", &n);
for(ll i = 0; i < n; ++i) {
scanf("%s", str);
t.insert(str, strlen(str));
}
scanf("%d", &m);
for(ll i = 0; i < m; ++i) {
scanf("%s", str);
ll ret = t.find(str, strlen(str));
if(ret > 1) {
printf("REPEAT\n");
} else if(ret == 1) {
printf("OK\n");
} else {
printf("WRONG\n");
}
}
return 0;
}