浅谈Trie树

时间:2021-07-03 05:09:32

Trie树,也叫字典树。顾名思义,它就是一个字典

字典是干什么的?查找单词!(英文字典哦)

个人认为字典树这个名字起得特别好,因为它真的跟字典特别像,一会r你就知道了。

注:trie的中文翻译就是单词查找树

一、引入

先来看一个题:

给你n个单词构成一个字典,再给你一个单词,问此单词在字典中有没有出现。

简单,暴力!

时间复杂度:n*单词长度

再来看一个题:

给你n个单词构成一个字典,再给你m个单词,问这m个单词在字典中有没有出现。

再暴力!

时间复杂度:n*单词长度+m*n*单词长度

n≤1e4,m≤1e4,单词长度≤1e3

boom!

(╯‵□′)╯︵┻━┻

所以,为什么要学习使用Trie树?

因为它快!

二、概念

我们首先来看看Trie树是啥

浅谈Trie树

↑就是它

我们来解剖一下,好好研究研究(Trie树:瑟瑟发抖)

首先,我们发现树的每条边上都有一个字母

这就是Trie树的样子,每条边上都有一个字母,每个顶点代表从根到该节点的路径所对应的字符串(根结点除外)

其次,有些节点是红色的,有些则不是

这是什么意思呢?

结点不是代表单词嘛

所以如果结点标红,代表该单词在字典中实际出现过

如果看不太懂,不要紧,继续往下

三、插入单词(构建Trie树)

这里进入正题。第一步,构建trie树

就好比你想要查找单词,首先得有字典才行吧

step1:初始化

Trie树为空,只包含一个孤零零的根结点

浅谈Trie树

step2:插入单词(注:这里假设所有单词仅由小写字母构成)

插入一个单词的步骤如下:

(1)对于Trie树,我们从根结点开始,设该节点为P;对于这个单词,我们从第一个字母开始,设此字符为s

(2)扫描P下方的所有边,看s有没有出现过

如果出现了,设s与P→Q这条边上的字符相同,则P=Q

如果没有出现,另建一条边,使该边上的字母为s,新节点为Q,然后P=Q

(3)s变为该单词的下一个字符,重复步骤2,直到扫描完整个单词为止


以概念中的那个图为例

首先我们要插入abc这个单词

(1)P为根结点,s为'a'

发现根结点下方无'a',所以新建一条边

浅谈Trie树

(2)P为a下方那个结点,s为'b'

发现P下方无'b',所以新建一条边

浅谈Trie树

(3)P为b下方那个结点,s为'c'

发现P下方无'c',所以新建一条边

浅谈Trie树

(4)发现abc这个单词插入完成,所以在当前的s做个标记,表示abc是一个出现过的单词

浅谈Trie树

接着我们按照相同的步骤插入bcd、efg和hij

浅谈Trie树

接下来插入abcd

(1)P为根结点,s为'a'

发现根结点下方有'a',所以P变为'a'下方的结点

(2)P为a下方那个结点,s为'b'

发现P下方有'b',所以P变为'b'下方的结点

(3)P为b下方那个结点,s为'c'

发现P下方有'c',所以P变为'c'下方的结点

(4)P为c下方那个红色的结点,s为'd'

发现P下方无'd',所以所以新建一条边

(5)单词abcd插入完毕,将当前的s做标记

浅谈Trie树

下面按照相同的方法,继续插入单词abd和b

插入完成后Trie树如下:

浅谈Trie树

看看一样不:

浅谈Trie树

(右下那条边应该是j)

我们可以发现,在Trie树中有相同前缀的单词共用相同的前缀,这样就可以大大的优化空间和时间

插入单词的时间复杂度为O(NE)(N为单词个数,E为单词长度)

参考代码:

void insert(char *s)//s为要插入的字符串
{
int len=strlen(s);
int u=;//1为根节点
for(int i=;i<len;i++)
{
int c=s[i]-'a';//'a'有时需换成'A'或'0'
if(!trie[u][c])//没有共同前缀,建立一个新的
ch[u][c]=++tot;//tot为总点数
u=ch[u][c];//继续向下插入单词
}
book[u]=true;//标记是一个出现过的单词(图中涂红色)
}

insert

四、查找单词

词典有了,接下来就可以查词了

在trie树中查单词就跟查字典一样。先查首字母,然后第二个,第三个……

查找过程跟插入的过程很像:

(1)对于Trie树,我们从根结点开始,设该节点为P;对于这个单词,我们从第一个字母开始,设此字符为s

(2)扫描P下方的所有边,看s有没有出现过

如果出现了,设s与P→Q这条边上的字符相同,则P=Q

如果没有出现,则该单词没有出现过,直接返回false

(3)s变为该单词的下一个字符,重复步骤2,直到扫描完整个单词为止

(4)扫描完成后,判断节点P有没有被标记(是不是某个出现过的单词的结尾)。如果标记了,那么该单词出现过,返回true;如果没有标记,那么该单词是词典中这个单词的前缀,返回false。

对于(4)讲解一下:

还是这棵Trie树:

浅谈Trie树

我们查找ef这个单词:

(1)P为根结点,s为'e'

发现根结点下方有'e',所以P变为'e'下方的结点

(2)P为e下方那个结点,s为'f'

发现P下方有'f',所以P变为'f'下方的结点

(3)查找完成,发现P这里没有标记,所以ef是词典中单词efg的前缀,而不是直接出现在了词典里,返回false

浅谈Trie树

再举一个例子,我们查找abd这个单词:

(1)P为根结点,s为'a'

发现根结点下方有'a',所以P变为'a'下方的结点

(2)P为a下方那个结点,s为'b'

发现P下方有'b',所以P变为'b'下方的结点

(3)P为b下方那个结点,s为'd'

发现P下方有'd',所以P变为'd'下方的结点

(3)查找完成,发现P这里有标记,所以abd是词典中的单词,返回true

应该讲的挺明白的

查找的一个单词的时间复杂度O(E),比起暴力的O(NE)要快多了

参考代码:

bool find(char *s)//s为要查找的字符串
{
int len=strlen(s);
int u=;//1为根节点
for(int i=;i<len;i++)
{
int c=s[i]-'a';//'a'有时需换成'A'或'0'
if(!trie[u][c])//单词没有出现,直接返回false
return false;
u=ch[u][c];//继续向下查找单词
}
//如果扫描完了这个单词
return true;//是某个单词的前缀
}

查找单词是否是词典中某单词的前缀

bool find(char *s)//s为要查找的字符串
{
int len=strlen(s);
int u=;//1为根节点
for(int i=;i<len;i++)
{
int c=s[i]-'a';//'a'有时需换成'A'或'0'
if(!trie[u][c])//单词没有出现,直接返回false
return false;
u=ch[u][c];//继续向下查找单词
}
//如果扫描完了这个单词
return book[u];//如果出现过,返回true;如果没有出现过(是前缀),返回false
}

查找单词是否在词典中出现过

模板题:

https://www.cnblogs.com/llllllpppppp/p/9366344.html

本文部分图片来源于网络

部分内容参考《信息学奥赛一本通.提高篇》第二部分第三章 Trie字典树

若需转载,请注明https://www.cnblogs.com/llllllpppppp/p/9449846.html

~祝大家编程顺利~