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

题意：给出n个模式串，给出m个查询串

 　　  问每一个查询串是否出现过？如果没出现过，就打印WRONG

　　　如果出现过，则看起被查询了几次，如果到目前为止（包括本次）查询过一次，则输出OK

　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　多次则输出 REPEAT

思路：直接建字典树

　　　跑查询的时候，如果找不到，就返回0

　　　　如果已经找到，则用vis数组来标记到目前为止这个串被查询了多少次

　　　　　　然后返回按对应结果输出

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;  //大小开模式串个数*模式串长度
char s[100];
int vis[maxn];
int trie[maxn][26];
int tot;
void Insert()
{
    int rt=0;
    for(int i=0;i<strlen(s);i++){
        int id=s[i]-'a';
        if(!trie[rt][id]) trie[rt][id]=++tot;
        rt=trie[rt][id];
    }
}
int Search()
{
    int rt=0;
    for(int i=0;i<strlen(s);i++){
        int id=s[i]-'a';
        if(!trie[rt][id]) return 0;
        rt=trie[rt][id];
    }  //rt经过此循环后变成前缀最后一个字母所在位置
    vis[rt]++;
    return vis[rt]; //返回当前字符串结尾节点的访问次数，也就是作为前缀的出现次数
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        Insert();
    }
    int m;
    scanf("%d",&m);
    while(m--){
        scanf("%s",s);
        int tmp=Search();
        if(!tmp) printf("WRONG\n");
        else if(tmp==1) printf("OK\n");
        else printf("REPEAT\n");
    }
    return 0;
}