[洛谷 2580] 于是他错误的点名开始了 (字典树)

题目背景

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)]='\0';        insert(s);//建树    }    int m;    scanf("%d",&m);    for(int i=1;i<=m;i++){        scanf("%s",s);        s[strlen(s)]='\0';//搜索        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;}