标签:sea insert 产生 blog alt mda 作用 问题: 没有
学习博客:https://blog.csdn.net/SongBai1997/article/details/82317259
这篇博客讲的挺好的,很详细,很好理解。
Trie树(字典树)
一、引入
字典是干啥的?查找字的。
字典树自然也是起查找作用的。查找的是啥?单词。
看以下几个题:
1、给出n个单词和m个询问,每次询问一个单词,回答这个单词是否在单词表中出现过。
答:简单!map,短小精悍。
好。下一个
2、给出n个单词和m个询问,每次询问一个前缀,回答询问是多少个单词的前缀。
答:map,把每个单词拆开。
judge:n<=200000,TLE!
这就需要一种高级数据结构——Trie树(字典树)
二、原理
在本篇文章中,假设所有单词都只由小写字母构成
对cat,cash,app,apple,aply,ok 建一颗字典树,建成之后如下图所示
由此可以看出:
1、字典树用边表示字母
2、有相同前缀的单词公用前缀节点,那我们可以的得出每个节点最多有26个子节点(在单词只包含小写字母的情况下)
3、整棵树的根节点是空的。为什么呢?便于插入和查找,这将会在后面解释。
4、每个单词结束的时候用一个特殊字符表示,图中用的‘′,那么从根节点到任意一个‘′,那么从根节点到任意一个‘’所经过的边的所有字母表示一个单词。
三、基本操作
A、insert,插入一个单词
1.思路
从图中可以直观看出,从左到右扫这个单词,如果字母在相应根节点下没有出现过,就插入这个字母;否则沿着字典树往下走,看单词的下一个字母。
这就产生一个问题:往哪儿插?计算机不会自己选择位置插,我们需要给它指定一个位置,那就需要给每个字母编号。
我们设数组trie[i][j]=k,表示编号为i的节点的第j个孩子是编号为k的节点。
什么意思呢?
这里有2种编号,一种是i,k表示节点的位置编号,这是相对整棵树而言的;另一种是j,表示节点i的第j的孩子,这是相对节点i而言的。
不理解?看图
还是单词cat,cash,app,apple,aply,ok
我们就按输入顺序对其编第一种号,红色表示编号结果。因为先输入的cat,所以c,a,t分别是1,2,3,然后输入的是cash,因为c,a是公共前缀,所以从s开始编,s是4,以此类推。
注意这里相同字母的编号可能不同
第二种编号,相对节点的编号,紫色表示编号结果。
因为每个节点最多有26个子节点,我们可以按他们的字典序从0——25编号,也就是他们的ASCLL码-a的ASCLL码。
注意这里相同字母的编号相同
实际上每个节点的子节点都应该从0编到——25,但这样会发现许多事根本用不到的。比如上图的根节点应该分出26个叉。节约空间,用到哪个分哪个。
这样编号有什么用呢?
回到数组trie[i][j]=k。 数组trie[i][j]=k,表示编号为i的节点的第j个孩子是编号为k的节点。
那么第二种编号即为j,第一种编号即为i,k
2、代码
void Insert(char s[maxn],int rt) { int len=strlen(s); for(int i=0;i<len;i++) { int x=s[i]-‘a‘; if(trie[rt][x]==0)//现在插入的字母在之前同一节点未出现过 { trie[rt][x]=++tot;//字母插入一个新的位置 如出现过则不用再存了 } rt=trie[rt][x];//为下一个字母插入做准备 } isw[rt]=true;//标记这个是单词 }
B、search,查找
查找有很多种,可以查找某一个前缀,也可以查找整个单词。
再次我们以查找一个单词是否出现过为例讲解
1、思路
从左往右以此扫描每个字母,顺着字典树往下找,能找到这个字母,往下走,否则结束查找,即没有这个单词;扫完了,判断是否有这个单词就行了。
2、代码
bool Find(char s[maxn],int rt) { int len=strlen(s); for(int i=0;i<len;i++) { int x=s[i]-‘a‘; if(trie[rt][x]==0) return false; rt=trie[rt][x];//为查询下一个字母做准备 } return isw[rt];//为啥需要这个呢? 因为你到了这里未必就有这个单词 可能只是其他单词的一个前缀罢了 }
完整代码:单词是否出现过:
#include<iostream> #include<cstdio> #include<cstring> using namespace std; const int maxn=1e6+5; int trie[maxn][30]; bool isw[maxn]; int tot=0; void Insert(char s[maxn],int rt) { int len=strlen(s); for(int i=0;i<len;i++) { int x=s[i]-‘a‘; if(trie[rt][x]==0)//现在插入的字母在之前同一节点未出现过 { trie[rt][x]=++tot;//字母插入一个新的位置 如出现过则不用再存了 } rt=trie[rt][x];//为下一个字母插入做准备 } isw[rt]=true;//标记这个是单词 } bool Find(char s[maxn],int rt) { int len=strlen(s); for(int i=0;i<len;i++) { int x=s[i]-‘a‘; if(trie[rt][x]==0) return false; rt=trie[rt][x];//为查询下一个字母做准备 } return isw[rt];//为啥需要这个呢? 因为你到了这里未必就有这个单词 可能只是其他单词的一个前缀罢了 } int main() { int rt=1; int N;//有N个单词 scanf("%d",&N); char s[maxn]; for(int i=1;i<=N;i++) { scanf("%s",s); Insert(s,rt); } scanf("%d",&N);//查询N个单词是否出现过 for(int i=1;i<=N;i++) { scanf("%s",s); if(Find(s,rt)) printf("YES\n"); else printf("NO\n"); } return 0; }
标签:sea insert 产生 blog alt mda 作用 问题: 没有
原文地址:https://www.cnblogs.com/caijiaming/p/11140069.html