洛谷 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总是送分的。

二、算法分析说明

字典树（Trie）又称前缀树，是一种针对字符串匹配优化的数据结构。其根节点为空，每个节点都有若干指针指向下一个节点。按字符串查询时，从根节点开始（已匹配 0 位），每深入一层代表已匹配待查找字符串多一位。


（两种画法是等价的）
字典树携带的信息可以保存在中间的非叶节点上，也可以保存在叶节点上。
本题给出的代码将字典树封装成一个类。虽然每个节点都有一个 unsigned 变量 visited 代表查询的名字是否已经点过，但是只有叶节点上的 visited 变量是具有实际意义的。

三、AC 代码

法一 unordered_map
该方法的耗时比字典树要多。

#include<iostream>
#include<unordered_map>
#include<string>
#pragma warning(disable:4996)
using namespace std;
unsigned m, n; unordered_map<string, bool> f; pair<string, bool> p; string name;
int main() {
	cin >> n; ++n; p.first.resize(50);
	while (--n) {
		cin >> p.first; f.emplace(p);
	}
	cin >> m; ++m; unordered_map<string, bool>::iterator I;
	while (--m) {
		cin >> name; I = f.find(name);
		if (I == f.end())puts("WRONG");
		else if (I->second == false) { puts("OK"); I->second = true; }
		else puts("REPEAT");
	}
	return 0;
}

法二 字典树
本方法中如果将map替换成unordered_map，耗时会增加。原因是unordered_map虽然具有O(1)的复杂度，但是常数较大。
最上面和最下面的提交将字典树的map换成unordered_map，中间则用字典树做但保持map不变。

#include<cstdio>
#include<map>
#pragma warning(disable:4996)
class trie {
private:
	struct node { std::map<char, node*> next; unsigned visited = 0; };
	std::map<char, node*>::iterator I; std::pair<char, node*> P;
	node* root;
public:
	trie() { root = new node(); }
	void insert(const char* const k) {
		node* p = root; const char* q = k;
		while (*q != 0) {
			I = p->next.find(*q);
			if (I == p->next.end()) { P.first = *q; P.second = new(node)(); p->next.emplace(P); I = p->next.find(*q); }
			p = I->second, ++q;
		}
	}
	unsigned find(const char* const w) {
		node* p = root; const char* q = w;
		while (*q != 0) {
			I = p->next.find(*q);
			if (I == p->next.end()) { return 2; }
			p = I->second, ++q;
		}
		if (p->visited == 0) { p->visited = 1; return 0; }
		else return 1;
	}
};
trie t; unsigned m, n; char name[51];
int main() {
	scanf("%u", &n);
	for (unsigned i = 0; i < n; ++i) { scanf("%s", name); t.insert(name); }
	scanf("%u", &m);
	for (unsigned i = 0; i < m; ++i) {
		scanf("%s", name);
		switch (t.find(name)) {
		case 0:puts("OK"); continue;
		case 1:puts("REPEAT"); continue;
		default:puts("WRONG");
		}
	}
	return 0;
}