P2580 于是他错误的点名开始了（trie字典树）

题目背景

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

本题尝试过暴力解，但不出所料TLE了。所以特地学了tried。下面直接上代码。

#include<stdio.h>
#include<string.h>
struct node
{
    int a;//该点序号；
    int b;//标记是否重复出现；
    int c;//单词结尾标记；
    int next[26];//每层直接开26个节点；
};
struct node tree[10000000];
char s1[51];
int total=0;//记录节点总数；
void insert(char *s1);//插入函数；
int find(char *s1);//查找函数；
int main()
{
    int n,m;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%s",s1);
        insert(s1);
        memset(s1,0,sizeof(s1));
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
        scanf("%s",s1);
        int d=find(s1);
        if(d==0)
        printf("WRONG\n");
        if(d==1)
        printf("REPEAT\n");
        if(d==2)
        printf("OK\n");
        memset(s1,0,sizeof(s1));
    }
    return 0;
}
void insert(char *s1)
{
    int len;
    len=strlen(s1);
    int last=0;
    for(int i=0;i<len;i++)
    {
        if(tree[last].next[s1[i]-'a']==0)//该字母没有入树；
        {
            total++;
            tree[last].a=total;
            tree[last].next[s1[i]-'a']=total;
            last=total;
        }
        else//该字母已经入树；
        last=tree[last].next[s1[i]-'a'];//搜索下一个节点；
        if(i==len-1)
        tree[last].c=1;
    }
}
int find(char *s1)
{
    int len;
    len=strlen(s1);
    int last=0;
    for(int i=0;i<len;i++)
    {
        if(tree[last].next[s1[i]-'a']==0)//树中没有出该字符串；
        return 0;
        last=tree[last].next[s1[i]-'a'];//继续搜索下一个节点；
    }
    if(tree[last].c==0)
    return 0;
    if(tree[last].b==1)
    return 1;
    tree[last].b=1;
    return 2;
}