洛谷P2580 於是他錯誤的點名開始了（字典樹trie）

題目背景

XS中學化學競賽組教練是一個酷愛爐石的人。

他會一邊搓爐石一邊點名以至於有一天他連續點到了某個同學兩次，然後正好被路過的校長髮現了然後就是一頓歐拉歐拉歐拉（詳情請見已結束比賽CON900）。

題目描述

這之後校長任命你爲特派探員，每天記錄他的點名。校長會提供化學競賽學生的人數和名單，而你需要告訴校長他有沒有點錯名。（爲什麼不直接不讓他玩爐石。）

輸入輸出格式

輸入格式：

 

第一行一個整數 n，表示班上人數。接下來 n 行，每行一個字符串表示其名字（互不相同，且只含小寫字母，長度不超過 50）。第 n+2 行一個整數 m，表示教練報的名字。接下來 m 行，每行一個字符串表示教練報的名字（只含小寫字母，且長度不超過 50）。

 

輸出格式：

 

對於每個教練報的名字，輸出一行。如果該名字正確且是第一次出現，輸出“OK”，如果該名字錯誤，輸出“WRONG”，如果該名字正確但不是第一次出現，輸出“REPEAT”。（均不加引號）

 

輸入輸出樣例

輸入樣例#1： 

5  
a
b
c
ad
acd
3
a
a
e

輸出樣例#1： 

OK
REPEAT
WRONG

說明

對於 40%的數據，n≤1000,m≤2000；

對於 70%的數據，n≤10000,m≤20000；

對於 100%的數據， n≤10000，m≤100000。

T1總是送分的。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
int sum[1000001][40],visit[1000001];
char a[100055];
int ans;
int num=1;
void in(char* s) {
    int len=strlen(s);
    int wzx=1;
    for(int i=0; i<len; i++) {
        int ch=s[i]-'a';
        if(sum[wzx][ch]==0)
            sum[wzx][ch]=++num;
        wzx=sum[wzx][ch];
    }
    visit[wzx]=1;
}
int find(char* s) {
    int len=strlen(s);
    int wzx=1;
    for(int i=0; i<len; i++) {
        int ch=s[i]-'a';
        wzx=sum[wzx][ch];
        if(wzx==0)
            return 0;
    }
    ans=wzx;
    return visit[wzx];
}
int main() {
    int n,m;
    cin>>n;
    for(int i=1; i<=n; i++) {
        cin>>a;
        in(a);
    }
    cin>>m;
    for(int i=1;i<=m;i++) {
        cin>>a;
        int cc=find(a);
        if(cc==1) {
            printf("OK\n");
            visit[ans]++;
            continue;
        }
        if(cc>1) {
            printf("REPEAT\n");
            continue;
        }
        if(cc==0)
            printf("WRONG\n");
    }
    return 0;
}