标签:enter 搜索结果 lan name 字典序 速度 search 判断 tree
Trie树,又叫字典树、前缀树(Prefix Tree)、单词查找树 或 键树,是一种多叉树结构。如下图:
上图是一棵Trie树,表示了关键字集合{“a”, “to”, “tea”, “ted”, “ten”, “i”, “in”, “inn”} 。从上图可以归纳出Trie树的基本性质:
通常在实现的时候,会在节点结构中设置一个标志,用来标记该结点处是否构成一个单词(关键字)。
可以看出,Trie树的关键字一般都是字符串,而且Trie树把每个关键字保存在一条路径上,而不是一个结点中。另外,两个有公共前缀的关键字,在Trie树中前缀部分的路径相同,所以Trie树又叫做前缀树(Prefix Tree)。
Trie树的核心思想是空间换时间,利用字符串的公共前缀来减少无谓的字符串比较以达到提高查询效率的目的。
插入和查询的效率很高,都为$O(m)$,其中 $m$ 是待插入/查询的字符串的长度。
Trie树中不同的关键字不会产生冲突。
Trie树只有在允许一个关键字关联多个值的情况下才有类似hash碰撞发生。
Trie树不用求 hash 值,对短字符串有更快的速度。通常,求hash值也是需要遍历字符串的。
Trie树可以对关键字按字典序排序。
当 hash 函数很好时,Trie树的查找效率会低于哈希搜索。
空间消耗比较大。
检索/查询功能是Trie树最原始的功能。思路就是从根节点开始一个一个字符进行比较:
1 struct trie_node 2 { 3 bool isKey; // 标记该节点是否代表一个关键字 4 trie_node *children[26]; // 各个子节点 5 };
1 struct trie_node 2 { 3 int count; // 记录该节点代表的单词的个数 4 trie_node *children[26]; // 各个子节点 5 };
思路:为了实现词频统计,我们修改了节点结构,用一个整型变量count
来计数。对每一个关键字执行插入操作,若已存在,计数加1,若不存在,插入后count
置1。
注意:第一、第二种应用也都可以用 hash table 来做。
Trie树可以对大量字符串按字典序进行排序,思路也很简单:遍历一次所有关键字,将它们全部插入trie树,树的每个结点的所有儿子很显然地按照字母表排序,然后先序遍历输出Trie树中所有关键字即可。
例如:找出一个字符串集合中所有以ab
开头的字符串。我们只需要用所有字符串构造一个trie树,然后输出以a->b->
开头的路径上的关键字即可。
trie树前缀匹配常用于搜索提示。如当输入一个网址,可以自动搜索出可能的选择。当没有完全匹配的搜索结果,可以返回前缀最相似的可能。
如后缀树,AC自动机等。
这里为了方便,我们假设所有的关键字都由 a-z 的字母组成。下面是 trie 树的一种典型实现:
1 #include <iostream> 2 #include <string> 3 using namespace std; 4 5 #define ALPHABET_SIZE 26 6 7 typedef struct trie_node 8 { 9 int count; // 记录该节点代表的单词的个数 10 trie_node *children[ALPHABET_SIZE]; // 各个子节点 11 }*trie; 12 13 trie_node* create_trie_node() 14 { 15 trie_node* pNode = new trie_node(); 16 pNode->count = 0; 17 for(int i=0; i<ALPHABET_SIZE; ++i) 18 pNode->children[i] = NULL; 19 return pNode; 20 } 21 22 void trie_insert(trie root, char* key) 23 { 24 trie_node* node = root; 25 char* p = key; 26 while(*p) 27 { 28 if(node->children[*p-‘a‘] == NULL) 29 { 30 node->children[*p-‘a‘] = create_trie_node(); 31 } 32 node = node->children[*p-‘a‘]; 33 ++p; 34 } 35 node->count += 1; 36 } 37 38 /** 39 * 查询:不存在返回0,存在返回出现的次数 40 */ 41 int trie_search(trie root, char* key) 42 { 43 trie_node* node = root; 44 char* p = key; 45 while(*p && node!=NULL) 46 { 47 node = node->children[*p-‘a‘]; 48 ++p; 49 } 50 51 if(node == NULL) 52 return 0; 53 else 54 return node->count; 55 } 56 57 int main() 58 { 59 // 关键字集合 60 char keys[][8] = {"the", "a", "there", "answer", "any", "by", "bye", "their"}; 61 trie root = create_trie_node(); 62 63 // 创建trie树 64 for(int i = 0; i < 8; i++) 65 trie_insert(root, keys[i]); 66 67 // 检索字符串 68 char s[][32] = {"Present in trie", "Not present in trie"}; 69 printf("%s --- %s\n", "the", trie_search(root, "the")>0?s[0]:s[1]); 70 printf("%s --- %s\n", "these", trie_search(root, "these")>0?s[0]:s[1]); 71 printf("%s --- %s\n", "their", trie_search(root, "their")>0?s[0]:s[1]); 72 printf("%s --- %s\n", "thaw", trie_search(root, "thaw")>0?s[0]:s[1]); 73 74 return 0; 75 }
对于Trie树,我们一般只需要插入和搜索操作。上面这段代码实现了一棵Trie树,该Trie树可以用来检索单词和统计词频。
标签:enter 搜索结果 lan name 字典序 速度 search 判断 tree
原文地址:http://www.cnblogs.com/stranger-/p/7350233.html