最近遇到了几道有关trie的题目,我就来讲讲,顺便梳理梳理思路。
字典树
引入
什么是字典树?字典树有什么用?
举个栗子,我们这里有一堆字符串,我们要在里面找到某个字符串在不在这堆字符串里面,大家想想有什么办法?
一个很朴素的方法是 O ( n ⋅ l e n ) O(n·len) O(n⋅len)的时间复杂度去一个个枚举,可是当有多组询问且 n n n, l e n len len 很大时,就显得有点吃力了,那我们有没有什么其他的办法呢?
这就引入了我们的字典树Trie了
结构
很明显,字典树是一棵树(废话),但是字典树的节点是没有任何用处的,不过它的边值得就是一个字符,由此可得,字典树上一条路径的边的值就是一个字符串。
配合图片食用更加:
假如说我们要插入 a b c abc abc , a b d abd abd , c b a cba cba 这三个字符串的话,插入后的样子是这样的:
五角星是一个标记,表示单词的某尾。
一般来说,字典树的构建以及查询都是不用递归的,具体方法在待会儿例题中会讲到。
例题
题目描述
新学期开学有几天了,Capo发现班上有一个老师,这位老师喜欢每天上课前先点名,并且是随机抽点。今天,Capo突然发现自己被点到了两次,于是Capo开始质疑老师的点名是否有重复或误报为其他班的同学。当然Capo可不想一个个比较,所以他把这个任务交给了你,当然Capo会提供班上人数n和他们的名字,同时Capo也会记下老师报的名字。
输入
第一行一个整数n,表示班上人数。接下来n行,每行一个字符串表示其名字(互不相同,且只含小写字母,长度不超过60)。
第n+2行一个整数m,表示老师报的名字。接下来m行,每行一个字符串表示老师报的名字(只含小写字母,且长度不超过60)。
输出
对于每个老师报的名字,输出一行。如果该名字正确且是第一次出现,输出"OK",如果该名字错误,输出"WRONG",如果该名字正确但不是第一次出现,输出"REPEAT"。(均不加引号)
样例输入
5
a
b
c
ad
acd
3
a
a
e
样例输出
OK
REPEAT
WRONG
数据范围限制
40%的数据:n≤1000, m≤2000;
70%的数据:n≤10000, m≤20000;
100%的数据: n≤10000,m≤100000。
思路
很明显,这是一道trie的模板题,只用在查找时找到字符串后打上个标记即可。
#include<string>
#include<cstdio>
using namespace std;
string s;
int n,m,k,trie[500010][26],bz[500010],bj[500010];
string read()
{
string a;
char ch=getchar();
while (ch<'a'||ch>'z') ch=getchar();
while (ch>='a'&&ch