作用
字典树是用于字符串快速检索的多叉树,每个节点都包含多个自负指针,将从根节点到某一个节点的路径上经过的字符链接起来,为该节点对应的字符串
字典树的创建
void insert(string s){ //将字符串s插入字典树中
int len=s.length(),p=1;
for(int i=0;i<len;i++){
int ch=s[i]-'a'; //转换为数字
if(!trie[p][ch])
trie[p][ch]=++tot;
p=trie[p][ch];
}
end[p]=true; //标记单词结束
}
字典树查找
bool search(string s){ //在字典树中查找该字符是否存在
int len=s.length(),p=1;
for(int i=0;i<len;i++){
p=trie[p][s[i]-'a'];
if(!p)
return false;
}
return true;
}
应用
1.字符串检索。事先将已知的一些字符串的有关信息存储到trie树里,查找字符串是否出现过
2.前缀统计。
3.最长公共前缀
4.排序
题目
1.单词翻译
思路
本题数据量大,采用暴力搜索会超时。可以将英语单词存储到数组中,将外语单词存储到字典树中,并记录该外语对应的英语单词的下标。查询时在字典树中查找信息中的每一个外语单词,若查找成功,则返回对应单词的下标
void insert(string s,int k) //将字符串s插入字典树中
{
int len=s.length(),p=1;
for(int i=0;i<len;i++){
int ch=s[i]-'a'; //转化为数字
if(!trie[p][ch])
trie[p][ch]=++tot; //记录下标
p=trie[p][ch];
}
value[p]=k; //记录下标
end[p]=1;
}
int query(string s)//查询
{
int len=s.length(),p=1;
for(int i=0;i<len;i++)
{
int ch=s[i]-'a'; //转换为数字
p=trie[p][ch];
if(!p)
return 0;
}
if(end[p])
return value[p]; //返回下标
return 0;
}
2.统计难题
思路
1.将字符串依次插入字典树中,用cnt[p]统计经过p节点的单词数
2.在字典树中查询字符串,返回最后一个节点的cnt[],为以该节点为前缀的单词的数量
void insert(string s) //将字符串s插入字典树中
{
int len=s.length(),p=1;
for(int i=0;i<len;i++)
{
int ch=s[i]-'a'; //转换为数字
if(!trie[p][ch])
trie[p][ch]=++tot; //记录下表
p=trie[p][ch];
cnt[p]++;
}
}
int query(string s) //查询
{
int len=s.length(),p=1;
for(int i=0;i<len;i++)
{
int ch=s[i]-'a'; //转换为数字
p=tire[p][ch];
if(!p)
return 0;
}
return cnt[p];
}