洛谷P2580 于是他错误的点名开始了

P2580 于是他错误的点名开始了

题目背景

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

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

题目描述

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

输入格式

第一行一个整数 \(n\)，表示班上人数。

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

第 \(n+2\) 行一个整数 \(m\)，表示教练报的名字个数。

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

输出格式

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

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

输入输出样例 #1

输入 #1

5  
a
b
c
ad
acd
3
a
a
e

输出 #1

OK
REPEAT
WRONG

说明/提示

• 对于 \(40\%\) 的数据，\(n\le 1000\)，\(m\le 2000\)。
• 对于 \(70\%\) 的数据，\(n\le 10^4\)，\(m\le 2\times 10^4\)。
• 对于 \(100\%\) 的数据，\(n\le 10^4\)，\(m≤10^5\)。

---

\(\text{upd 2022.7.30}\)：新增加一组 Hack 数据。

P2580 于是他错误的点名开始了

---

这题也要用到字典树
注：本题的讲解我将建立在上一篇题解的基础上
建议没看过我上一篇题解的博友以及不熟悉该知识点的博友请移步至：
https://www.cnblogs.com/AAAjiancaipifa/p/20045696
查看我的前一篇题解，关于字典树的详解，此处不过多赘述

题目解析

---

数据的定义与上一题一样：

const int N=1500000;
//唯一的问题就是这题数组开太小了会报错，我就这样一直卡在数据里......
string s[N];
int f[N],root,tot,n;
int trie[N][26];

唯一需要改变的是，这题的f数组不仅仅代表了结尾节点：

若p是结尾节点，f[p]=1

还将代表重复的字符串

若以p为结尾节点的字符串已经出现过,f[p]=2

以及附带着答案函数的改变:

int query(string s)
{
	int p=root;
	for(auto c:s)
	{
		p=trie[p][c-'a'];
	}
	
	//前面照旧
	
	if(f[p]==0) return 0; //如果f[p]=0,那么就没有这个人名，返回0
	if(f[p]==2) return 2; //如果f[p]=2,说明这个人名重复了，返回2
	f[p]=2;
	/*
	注意注意！！！！这里f[p]=1时并不是直接返回1，而是先赋值为2，因为f[p]这次找到了
	下次再找到相同的就是重复了，因此要提前赋值为2，这也是为什么我不直接return f[p]的原因
	*/
	return 1; //最后才是return 1
}

其他的部分只有微微的改动，代码附上：

ACcode:

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

const int N=1500000;
string s[N];
int f[N],root,tot,n;
int trie[N][26];
void ins(string s)
{
	int p=root;
	for(auto c:s)
	{
		if(trie[p][c-'a']==0) trie[p][c-'a']=++tot;
		p=trie[p][c-'a'];
	}
	f[p]=1;
}
int query(string s)
{
	int p=root;
	for(auto c:s)
	{
		p=trie[p][c-'a'];
	}
	if(f[p]==0) return 0;
	if(f[p]==2) return 2;
	f[p]=2;
	return 1;
}
int main()
{
	ios::sync_with_stdio(0);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>s[i];
		ins(s[i]);
	}
	int q;
	cin>>q;
	while(q--)
	{
		string t;
		cin>>t;
		int tmp=query(t);
		//这里也要提前记录，因为query函数一旦调用,f[p]的值可能就会发生改变（例如，从f[p]=1->f[p]=2）
		if(tmp==0) cout<<"WRONG"<<endl;
		else if(tmp==1) cout<<"OK"<<endl;
		else cout<<"REPEAT"<<endl;
		//然后结合题意去输出就好了
	}
	return 0;
}
//Author:AAA_jiancaipifa

Bye~~~~~~~