trie树(字典树)

Posted bennettz

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了trie树(字典树)相关的知识,希望对你有一定的参考价值。

核心思想:

利用字符串的公共前缀来降低查询时间的开销以达到提高效率的目的

举个例子

上图是由

  • am
  • as
  • tea
  • too
  • tooth
  • two

构成的字典树。每个节点代表的单词是从根遍历到他的路径,标黄的是当前节点存在单词

代码实现:

struct Trie{
    Trie *son[26];
    bool w;//记录当前节点是否是单词;
    Trie(){
        memset(son,0,sizeof(son)),w=0;
    } 
}root;

基本性质:

  1. 根节点不包含字符,除根节点外每一个节点都只包含一个字符。
  2. 从根节点到某一节点,路径上经过的字符连接起来,为该节点对应的字符串。
  3. 每个节点的所有子节点包含的字符都不相同。

查询和插入

对于一个单词,我只要按照它的每个字母从根走到对应的节点,再看这个节点是否被标记为黄色就可以知道它是否出现过了。
如果没被标记,把这个节点标记为红色,就相当于插入了这个单词,这样查找和插入就可以一起实现。

代码实现:(这里字典树只支持大写字母)

int  sum,word;
void search(char s[]){//查找和插入 
    Trie *x=&root;
    for(int i=0;s[i];i++){
        if(x->son[s[i]-\'A\']==NULL){//判断之前是否出现过 
            x->son[s[i]-\'A\']=new Trie;//创建新的节点 
            sum++;//sum总结点数+1
        }
        x=x->son[s[i]-\'A\'];
    }
    if(!x->w)word,x->w=1;//word总单词数+1 
}

因为空间是动态分配的,所以用完要释放

void node_free(Trie* x){//节点释放 
    for(int i=0;i<26;i++){
        if(x->son[i]!=NULL)node_free(x->son[i]);//先释放子节点 
    }
    delete x;
} 
void trie_free(){//空间释放 
    for(int i=0;i<26;i++){
        if(root.son[i]!=NULL)node_free(root.son[i]),root.son[i]=NULL;
    }
}

 例题

[NOI2000] 单词查找树

代码

#include<cstdio>
#include<cstring>
using namespace std;
struct Trie{
    Trie *son[26];
    bool w;//记录当前节点是否是单词;
    Trie(){
        memset(son,0,sizeof(son)),w=0;
    } 
}root;
int  sum,word;
void search(char s[]){//查找和插入 
    Trie *x=&root;
    for(int i=0;s[i];i++){
        if(x->son[s[i]-\'A\']==NULL){//判断之前是否出现过 
            x->son[s[i]-\'A\']=new Trie;//创建新的节点 
            sum++;
        }
        x=x->son[s[i]-\'A\'];
    }
    if(!x->w)word,x->w=1;//sum总结点数+1 
}
void node_free(Trie* x){//节点释放 
    for(int i=0;i<26;i++){
        if(x->son[i]!=NULL)node_free(x->son[i]);//先释放子节点 
    }
    delete x;
} 
void trie_free(){//空间释放 
    for(int i=0;i<26;i++){
        if(root.son[i]!=NULL)node_free(root.son[i]);
    }
}
char s[64];
int main(){
    freopen("trie.in","r",stdin);
    freopen("trie.out","w",stdout);
    while(~scanf("%s",s)){
        search(s);
    }
    trie_free();
    printf("%d",sum+1);
    return 0;
}

 

以上是关于trie树(字典树)的主要内容,如果未能解决你的问题,请参考以下文章

trie树(字典树)

前缀树(字典树/Trie) -----Java实现

数据结构——trie树(字典树)

Trie树(字典树)

LeetCode 208. 实现 Trie (前缀树)

01字典树