题目描述
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
说明
对于 40%的数据,n≤1000,m≤2000;
对于 70%的数据,n≤10000,m≤20000;
对于 100%的数据, n≤10000,m≤100000。
分析
根据字典树,我们可以建e数组表是否为名字结尾,建same数组表该名字是否第一次被点到
然后学会字典树就行了(滑稽)
#include <iostream>
#include <cstdio>
#include <cstring>
#define rep(i,a,b) for (i=a;i<=b;i++)
using namespace std;
int f[500001][26];
bool e[500001],same[500001];
int n,m;
int root=1,cnt;
void insert(char *s)
{
int a,r=root,i;
for (i=0;s[i];i++)
{
a=s[i]-'a';
if (!f[r][a])
f[r][a]=++cnt;
r=f[r][a];
}
e[r]=1;
}
int sear(char *s)
{
int a,r=root,i;
for (i=0;s[i];i++)
{
a=s[i]-'a';
if (!f[r][a]) return 0;
r=f[r][a];
}
if (e[r])
if (!same[r])
return same[r]=1;
else
return 2;
}
void init()
{
int i;
char a[51];
scanf("%d",&n);
rep(i,1,n)
{
scanf("%s",&a);
insert(a);
}
}
void print()
{
int i,x;
char a[51];
scanf("%d",&m);
rep(i,1,m)
{
scanf("%s",&a);
x=sear(a);
if (!x)
printf("WRONG\n");
if (x)
if (x==1)
printf("OK\n");
else
printf("REPEAT\n");
}
}
int main()
{
init();
print();
}