于是他错误的点名开始了（字典树Trie模板）

原题链接

题目描述

输入描述

输出描述

输入样例

5  
a
b
c
ad
acd
3
a
a
e

输出样例

OK
REPEAT
WRONG

---

题目大意：对输入的 n 个名字建 Trie 树，并在 m 个询问内查找是否有对应的 Trie 树，同时注意是否为第一次询问该名字。

快乐模板题，唯一在建树和查找之上增加的内容是判断当某个 Trie 树存在时，标记它是否已经访问过。可用 1 表示第一次找到，2 表示已找到过一次即可。

参考代码

#include <bits/stdc++.h>
using namespace std;

const int N = 5e5+10;
char s[60];
int ch[N][26],st[N];
int cnt = 1;

int main() {
	int n,m;
	cin>>n;

	for(int i=1; i<=n;++i){
    	scanf("%s",s+1);
    	
		int p=1;
    	for(int j=1;s[j];++j){
      		int c=s[j]-'a';
      		if(!ch[p][c]) 
				ch[p][c]=++cnt;
      		p=ch[p][c];
    	}
    	st[p] = 1;
  	}

  	cin>>m;

 	while (m--) {
    	scanf("%s", s + 1);
    	
		int p = 1;
    	for(int j=1;s[j];++j) {
        	int c=s[j]-'a';
    		p=ch[p][c];
      		if(!p) 
			  	break;
    	}
    	
		if(st[p]==1){
      		st[p]=2;
      		cout<<"OK"<<endl;
    	}
		else if(st[p]==2)
    		cout<<"REPEAT"<<endl;
    	else
      		cout<<"WRONG"<<endl;
  	}

  return 0;
}