[洛谷 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;
}