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

这道题是字典树的模板题
传送门

题目

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

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

输出格式
对于每个教练报的名字，输出一行。
如果该名字正确且是第一次出现，输出 OK，如果该名字错误，输出 WRONG，如果该名字正确但不是第一次出现，输出 REPEAT。
输入输出样例
输入

5  
a
b
c
ad
acd
3
a
a
e

输出

OK
REPEAT
WRONG

解题思路

就是个模板题。建字典树要注意。还有要注意end[]数组的使用。在insert()中是写入时候用的。在query()中是查询结尾同时记录这个串出现了多少次。

#include<bits/stdc++.h>
using namespace std;
#define FRE freopen("1.txt", "r", stdin)
#define IOS ios::sync_with_stdio(false)
const int N = 5e5 + 10;

int tr[N][26], tot = 0;
int End[N];
void insert(string s) {
	int pos = 0;
	for(int i = 0; i < s.length(); i++) {
		int c = s[i] - 'a';
		if(tr[pos][c] == 0) {
			tr[pos][c] = ++tot;
		}
		pos = tr[pos][c];
	}
	End[pos] = 1;
	return;
}


int query(string s){
	int pos = 0;
	for(int i = 0; i < s.length(); i++){
		int c = s[i] - 'a';
		if(tr[pos][c] == 0){
			return 0;
		}
		pos = tr[pos][c];
	}
	if(End[pos] == 1){
		End[pos] = 2;
		return 1;
	}else if(End[pos] == 0){
		return 0;
	}
	return 2;
}

int main(int argc, char** argv){
	IOS;
	int n; cin >> n;
	for(int i = 0; i < n; i++) {
		string s; cin >> s;
		insert(s);
	}
	int m; cin >> m;
	for(int i = 0; i < m; i++){
		string s; cin >> s;
		int p = query(s);
		if(p == 0){
			cout << "WRONG" << endl;
		}else if(p == 1){
			cout << "OK" << endl;
		}else{
			cout << "REPEAT" << endl;
		}
	}
	return 0;
}