[洛谷P2580]于是他错误的点名开始了（Trie树）

传送门

洛谷P2580的一个水题，用啥都能过，不过为了练习一下刚刚学会的字典树，还是认真做一下吧。

#include <cstdio>
#include <cstring>

using namespace std;

#define idx(x) x - 'a'

int n, m, nex;
struct node
{
    int next[];
    int val;
}tree[];

void Insert(char *s)
{
    int i, rt = , len = strlen(s) - ;
    for(i = ; i <= len; i++)
    {
        int c = idx(s[i]);
        if(!tree[rt].next[c]) tree[rt].next[c] = ++nex;
        rt = tree[rt].next[c];
    }
    tree[rt].val = ;
}

int Find(char *s)
{
    int i, rt = , len = strlen(s) - ;
    for(i = ; i <= len; i++)
    {
        int c = idx(s[i]);
        if(!tree[rt].next[c]) return ;
        rt = tree[rt].next[c];
    }
    if(tree[rt].val == )
    {
        tree[rt].val++;
        return ;
    }
    else return ;
}

int main()
{
    int i;
    char str[];
    scanf("%d", &n);
    for(i = ; i <= n; i++)
    {
        scanf("%s", str);
        Insert(str);
    }
    scanf("%d", &m);
    for(i = ; i <= m; i++)
    {
        scanf("%s", str);
        int f = Find(str);
        if(f == ) printf("WRONG\n");
        else if(f == ) printf("OK\n");
        else printf("REPEAT\n");
    }
    return ;
}