洛谷P2580 于是他错误的点名开始了(Trie树板子)

字典树空间一般开n*maxLen(n为字符串数量，maxLen为字符串最大长度，最大情况n个串都不相同)

#include<bits/stdc++.h>
using namespace std;
const int maxnode = 5e5+5;//n*maxlen
const int maxn = 5e3+5;
int trie[maxnode][26],val[maxnode];
char str[maxn];
int rt,tot;//根节点永远是0且无值
void insert(){
    int pos=rt,len=strlen(str);
    for(int i=0;i<len;i++){
        int id = str[i]-'a';
        if(trie[pos][id]==0) trie[pos][id]=++tot;
        pos=trie[pos][id];
    }
    val[pos]=1;
}
int find(){
    int pos=rt,len=strlen(str);
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(trie[pos][id]==0) return 0;
        pos=trie[pos][id];
    }
    val[pos]++;
    return val[pos]-1;
}

int main(){
    int n,m;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%s",str);
        insert();
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%s",str);
        int f = find();
        if(f==0) printf("WRONG\n");
        else if(f==1) printf("OK\n");
        else printf("REPEAT\n");
    }
    return 0;
}