trie树及其应用

来源：互联网发布：气宗剑宗知乎编辑：程序博客网时间：2024/05/21 10:41

参考内容：

1. 这位童鞋的文章 http://blog.csdn.net/zhulei632/article/details/6704496

2. 严蔚敏 -数据结构

1.键树的定义：

键树又叫“数字查找树”。深度>=2 . 树中的每个节点一般不是直接包含关键字，而是包含组成关键字的符号（当然叶子节点除外，叶子节点可能包含整个单词以及词频，非叶节点也可包含单词和词频）。根据存储结构的不同，又分为双链树和多重链表树。或者就是常说的“Trie树”，取自检索“retrieve”中间的四个单词。因此也被称为检索树。Trie树的每个节点含有d个指针域（d为关键字的基数，如果是字母，那么基数为26 即a-z .如果是数字，那么基数是10, 即0-9）。

如图所示，一个Trie树的结构如下：

如果在trie树的node节点添加新的域 count,记录已有的单词总数。那么，Trie树除了实现单词查询之外，还可以实现单词频度统计。

如，我们定义的Trie树的节点结构如下：

[cpp] view plaincopyprint?
typedef struct Trie_node{  
    int  count;  
    struct Trie_node *next[26];  
  
}TrieNode, *Trie;  

其中next数组指向下一层次节点。

有了这个结构。Trie树的实现就有了一个基础。

2。Trie树的建立

建立一个Trie树的过程就是不断添加新的单词的过程。由根节点向下扫描，如果不存在相应的节点则创建之。否则进入下一个层次，直到单词添加完毕。

根据该算法，代码不难写出：

[cpp] view plaincopyprint?
//创建新的节点  
TrieNode* createTrieNode(){  
    TrieNode* root = (TrieNode*)malloc(sizeof(TrieNode));  
    root->count = 0;  
    memset(root->next, 0, sizeof(root->next));  
    return root;  
}  
//插入单词。  
void trie_insert(Trie root, char* word){  
    TrieNode* node = root;  
    char *p = word;  
    while(*p)  
    {  
        if(NULL == node->next[*p-'a'])  
        {  
            node->next[*p-'a'] = createTrieNode();  
        }  
        node = node->next[*p-'a'];  
        p++;  
    }  
    node->count += 1;  
}  

3 .Trie树的检索

Trie树中检索的过程是走一条从跟节点开始到叶子节点的路径（不一定走到叶子节点，取决于你的Trie树的实现，如果规定每个单词以$结束，那么检索成功的话一定走到叶子节点）：如下图示意,检索单词bat的路径用红色标出。

实现检索的代码：

[cpp] view plaincopyprint?
int trie_search(Trie root, char* word){  
    TrieNode* node = root;  
    char *p = word;  
    while(*p && node!=NULL)  
    {  
        node = node->next[*p-'a'];  
        p++;  
    }  
    return (node != NULL && node->count > 0);  
}  

4. 利用Trie树实现词频统计

如上文所述：在Trie的node节点中添加count域后，可以统计单词出现的次数。统计的方法就是在插入单词的时候，令相应的count域加1（初始化为0）。代码见Trie插入部分。

完整的测试代码如下：

[cpp] view plaincopyprint?
#include <stdio.h>  
#include <stdlib.h>  
#include <memory.h>  
  
typedef struct Trie_node{  
    int  count;  
    struct Trie_node *next[26];  
  
}TrieNode, *Trie;  
  
TrieNode* createTrieNode(){  
    TrieNode* root = (TrieNode*)malloc(sizeof(TrieNode));  
    root->count = 0;  
    memset(root->next, 0, sizeof(root->next));  
    return root;  
}  
  
void trie_insert(Trie root, char* word){  
    TrieNode* node = root;  
    char *p = word;  
    while(*p){  
        if(NULL == node->next[*p-'a']){  
            node->next[*p-'a'] = createTrieNode();  
        }  
        node = node->next[*p-'a'];  
        p++;  
    }  
    node->count += 1;  
}  
  
int trie_search(Trie root, char* word){  
    TrieNode* node = root;  
    char *p = word;  
    while(*p && node!=NULL){  
        node = node->next[*p-'a'];  
        p++;  
    }  
    return (node != NULL && node->count > 0);  
}  
  
int trie_word_count(Trie root, char* word){  
    TrieNode * node = root;  
    char *p = word;  
    while(*p &&node != NULL){  
        node = node->next[*p-'a'];  
        p++;  
    }  
    return node->count;  
}  
  
  
int main(){  
    Trie t = createTrieNode();  
    char word[][10] = {"test","study","open","show","shit","work","work","test","tea","word","area","word","test","test","test"};  
    for(int i = 0;i < 15;i++ ){  
        trie_insert(t,word[i]);  
    }  
    for(int i = 0;i < 15;i++ ){  
        printf("the word %s appears %d times in the trie-tree\n",word[i],trie_word_count(t,word[i]));  
    }  
    char s[10] = "testit";  
    printf("the word %s exist? %d \n",s,trie_search(t,s));  
    return 0;  
}  

运行结果如下：

0 0