题目背景
XS中学化学竞赛组教练是一个酷爱炉石的人。
他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次,然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉(详情请见已结束比赛CON900)。
题目描述
这之后校长任命你为特派探员,每天记录他的点名。校长会提供化学竞赛学生的人数和名单,而你需要告诉校长他有没有点错名。(为什么不直接不让他玩炉石。)
输入输出格式
输入格式:
第一行一个整数 n,表示班上人数。接下来 n 行,每行一个字符串表示其名字(互不相同,且只含小写字母,长度不超过 50)。第 n+2 行一个整数 m,表示教练报的名字。接下来 m 行,每行一个字符串表示教练报的名字(只含小写字母,且长度不超过 50)。
输出格式:
对于每个教练报的名字,输出一行。如果该名字正确且是第一次出现,输出“OK”,如果该名字错误,输出“WRONG”,如果该名字正确但不是第一次出现,输出“REPEAT”。(均不加引号)
输入输出样例
输入样例#1:
5
a
b
c
ad
acd
3
a
a
e
输出样例#1:
OK
REPEAT
WRONG
说明
对于 40%的数据,n≤1000,m≤2000;
对于 70%的数据,n≤10000,m≤20000;
对于 100%的数据, n≤10000,m≤100000。
T1总是送分的。
思路
简单的字典树题,维护一棵字典树(ch)
首先将录入的名单建成一棵字典树,在每个名字解释后标记(val)
再将点名的名单带入进行查询,再用一个数组(k)记录次单词被点名过没有
没有就输出OK,有就输出REPERT,(ps:我repeat拼错调了好久。。。)
代码如下
#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <cstdlib>
using namespace std;
char s[55];
int ch[500001][30];
int val[500001],k[500001];
int sz=1;//序号
void insert(char *s){//建树的函数
int u=0;//u当前节点
int len=strlen(s);
for(int i=0;i<len;i++){
int c=s[i]-'a';
if(ch[u][c]>0){
u=ch[u][c];
}
else
{
val[sz]=0;
ch[u][c]=sz;
u=sz;
sz++;
}
}
val[u]=1;//单词结尾标为1,标记他是个单词
}
int search(char *s){//
int u=0;
int len=strlen(s);
int flag=2;
int c;
for(int i=0;i<len;i++)
{
c=s[i]-'a';
if(!ch[u][c])//没有这个单词 返回 错误
return 2;
else u=ch[u][c];//有 就继续向下
}
//此时已到达搜索单词的最后一个字母
if(!val[u]) return 2;//如果不是单词(是其他单词的前缀)返回 失败
if(k[u]>0) return 1;//如果已经报过一遍了 返回 重复
k[u]=1;//标记 此单词 已经报过
return 0;//返回 正确
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",s);
s[strlen(s)]=' ';
insert(s);//建树
}
int m;
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%s",s);
s[strlen(s)]=' ';//搜索
int ans=search(s);//三种情况 分别输出
if(ans==0) cout<<"OK"<<endl;
else if(ans==1) cout<<"REPEAT"<<endl;
else if(ans==2) cout<<"WRONG"<<endl;
}
return 0;
}