Trie树

来源：互联网发布：淘宝汉服商家推荐2016 编辑：程序博客网时间：2024/06/05 21:51

Trie树是前缀树，是指的对于一个节点的所有子节点具有相同的前缀。通常使用在字符串检索，经典使用场景是在搜索提示中对用户搜索词的提示，根据用户当前输入的搜索词查看有那些词语的前缀和你的输入是一样的，参考wiki；可以用于在在切词分词中用来保存词库的数据结构。对一篇文章进行切词分词的时候，通过与词库对比查找得出要如何切词，切词分词的简单介绍。

和Trie相关的另一种数据结构就是后缀树。后缀树是具有相同后缀的所有节点组成的一个树状数据结构。

参考了wiki百科中的实现，对Trie树进行简单的实现。包括插入一个新的词语和搜索一个词语。其中对于建立一棵Trie树可以循环调用插入方法直到词库结束。对于搜索一个词其实是一个确定有限状态自动机的过程。后来看到wiki才想起在编译原理中学过的这个定义。简单来说就是给定一个状态之后，有一个确定的下一状态可以调整或者调整到结束状态。对于Tries树，每个节点其实就是一个状态，而节点之间的连线就是状态转移。

下面是代码实现，一个简单程序中c与c++风格都有，囧。

[cpp] view plaincopy

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define ALPHABET_SIZE 26
#define CHAR_TO_INDEX(c) ((int)c - (int)'a')
typedef struct trie_node{
int value;
trie_node *children[ALPHABET_SIZE];
};
typedef struct trie{
trie_node *root;
int count;
};
void initNode(trie_node *current_node){
for(int level = 0; level < ALPHABET_SIZE; level ++){
current_node -> children[level] = 0;
}
current_node -> value = -1;
}
void insert(trie *pTrie, char *key, int value){
if(pTrie == 0){
pTrie = (trie *) malloc(sizeof(trie));
pTrie -> count = 0;
}
pTrie -> count++;
trie_node *tn = pTrie -> root;
int length = strlen(key);
int level = 0, index = 0;
for(; level < length; level ++){
index = CHAR_TO_INDEX(key[level]);
if(tn -> children[index] == 0 && !(tn -> children[index])){
trie_node *new_node = (trie_node *) malloc(sizeof(trie_node));
initNode(new_node);
tn -> children[index] = new_node;
}
tn = tn -> children[index];
}
tn -> value = value;
}
int search(const trie *pTrie, char *key){
int result = -1;
if(pTrie == 0){
return result;
}
trie_node *tn = ((trie *) pTrie) -> root;
int level = 0, length = strlen(key);
int index = 0;
for(; level < length; level ++){
index = CHAR_TO_INDEX(key[level]);
if(tn -> children[index] == 0 || !(tn -> children[index])){
printf("can't finde the \" %s \"", key);
result = -1;
break;
}
tn = tn -> children[index];
result = level;
}
return result;
}
main(){
trie *pTrie = (trie *) malloc(sizeof(trie));
pTrie -> count = 0;
pTrie -> root = (trie_node *) malloc(sizeof(trie_node));
initNode(pTrie -> root);
trie_node *node = pTrie -> root;
insert(pTrie, "abc", -1);
int result = search(pTrie, "ac");
printf("\nresult num : %d\n", result);
system("pause");
}