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

传送门

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

#include <cstdio>
#include <cstring>

using namespace std;

#define idx(x) x - 'a'

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

void Insert(char *s)
{
int i, rt = 0, len = strlen(s) - 1;
for(i = 0; 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 = 1;
}

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

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