1. 题目

题目链接：P2580「于是他错误的点名开始了」 。

题目背景

XS中学化学竞赛组教练是一个酷爱炉石的人。

他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次，然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉（详情请见已结束比赛 CON900）。

题目描述

这之后校长任命你为特派探员，每天记录他的点名。校长会提供化学竞赛学生的人数和名单，而你需要告诉校长他有没有点错名。（为什么不直接不让他玩炉石。）

输入格式

第一行一个整数 nn，表示班上人数。

接下来 nn 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 5050）。

第 n+2n+2 行一个整数 mm，表示教练报的名字个数。

接下来 mm 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 5050）。

输出格式

对于每个教练报的名字，输出一行。

如果该名字正确且是第一次出现，输出 OK，如果该名字错误，输出 WRONG，如果该名字正确但不是第一次出现，输出 REPEAT。

输入输出样例

输入 #1

1
2
3
4
5
6
7
8
9
10
5  
a
b
c
ad
acd
3
a
a
e

输出 #1

1
2
3
OK
REPEAT
WRONG

说明/提示

  • 对于 40%40\% 的数据，n≤1000n \leq 1000，m≤2000m \leq 2000 。
  • 对于 70%70\% 的数据，n≤104n \leq 10^4，m≤2×104m \leq 2 \times 10^4 。
  • 对于 100%100\% 的数据，n≤104n \leq 10^4，m≤105m \leq 10^5 。

2. 题解

分析

定睛一看，字典树板子题，结束。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include <bits/stdc++.h>
using namespace std;

// 字典树
struct Trie {
    #ifndef _TRIE_
    #define ll int
    #define MAXN 500005
    #define MAXCHAR 26
    #endif
    ll cnt;                     // 动态开点（开 Trie 树结点编号）
    ll next[MAXN][MAXCHAR];     // 记录关联边为对应字符的子结点的下表
    ll exist[MAXN];             // 记录以该结点结尾的字符串是否存在
    Trie(): cnt(0) {}
    // 插入字符串
    void insert(char *str, ll len) {
        ll p = 0;
        for(ll i = 0; i < len; ++i) {
            ll c = str[i] - 'a';
            if(!next[p][c]) {
                next[p][c] = ++cnt;
            }
            p = next[p][c];
        }
        exist[p] = 1;
    }
    // 查找字符串
    ll find(char *str, ll len) {
        ll p = 0;
        for(ll i = 0; i < len; ++i) {
            ll c = str[i] - 'a';
            if(!next[p][c]) {
                return false;
            }
            p = next[p][c];
        }
        return exist[p] ? exist[p]++ : exist[p];
    }
};

int main()
{
    ll n, m;
    char str[MAXN];
    static Trie t;
    scanf("%d", &n);
    for(ll i = 0; i < n; ++i) {
        scanf("%s", str);
        t.insert(str, strlen(str));
    }
    scanf("%d", &m);
    for(ll i = 0; i < m; ++i) {
        scanf("%s", str);
        ll ret = t.find(str, strlen(str));
        if(ret > 1) {
            printf("REPEAT\n");
        } else if(ret == 1) {
            printf("OK\n");
        } else {
            printf("WRONG\n");
        }
    }
    return 0;
}