字典树模板题 于是他错误的点名开始了（洛谷 P2580）

于是他错误的点名开始了

题目描述

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

输入格式

第一行一个整数 n，表示班上人数。接下来 n 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 50）。第 n+2 行一个整数 m，表示教练报的名字。接下来 m 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 50）。

输出格式

对于每个教练报的名字，输出一行。如果该名字正确且是第一次出现，输出“OK”，如果该名字错误，输出“WRONG”，如果该名字正确但不是第一次出现，输出“REPEAT”。（均不加引号）

---

字典树模板题；

如果关闭了输入输出流的话，就不要用scanf，要不然会出错；

总的来说，字典树就是把线性问题转化为树上问题，降低复杂度；

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define lson k<<1
#define rson k<<1|1
using namespace std;
const int N=500100;
const int M=50100;
const LL mod=1e9+7;
int tree[N][30],rt,val[N];
void add(string s){//s为要插入的字符串
	int k=0;//根结点
	int len=s.length();
	for(int i=0;i<len;i++){
		int d=s[i]-'a';
		if(!tree[k][d]) tree[k][d]=++rt; 
		k=tree[k][d];//继续往下遍历 
	}
}
int search(string s){
	int k=0;
	int len=s.length();
	for(int i=0;i<len;i++){
		int d=s[i]-'a';
		if(!tree[k][d]) return 0;//wrong
		k=tree[k][d];
	}
	if(val[k]==0){
		val[k]=1;
		return 1;//ok
	}
	return 2;//repeat
} 
int main(){
	ios::sync_with_stdio(false);
	int n;
	cin>>n;
	string s;
	for(int i=1;i<=n;i++){
		cin>>s;
		add(s);
	}
	int m;
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>s;
		int flag=search(s);
		if(flag==0) cout<<"WRONG"<<endl;
		else if(flag==1) cout<<"OK"<<endl;
		else cout<<"REPEAT"<<endl;
	}
	return 0;
}