Trie树的C++实现
jopen
12年前
Trie—单词查找树
Trie,又称单词查找树、前缀树,是一种哈希树的变种。应用于字符串的统计与排序,经常被搜索引擎系统用于文本词频统计。
性质:
1.根节点不包含字符,除根节点外的每一个节点都只包含一个字符。
2.从根节点到某一节点,路径上经过的字符连接起来,为该节点对应的字符串。
3.每个节点的所有子节点包含的字符都不相同。
优点:
1.查询快。对于长度为m的键值,最坏情况下只需花费O(m)的时间;而BST需要O(m log n)的时间。
2.当存储大量字符串时,Trie耗费的空间较少。因为键值并非显式存储的,而是与其他键值共享子串。
操作:
1.初始化或清空:遍历Trie,删除所有节点,只保留根节点。
2.插入字符串
1).设置当前节点为根节点,设置当前字符为插入字符串中的首个字符;
2).在当前节点的子节点上搜索当前字符,若存在,则将当前节点设为值为当前字符的子节点;否则新建一个值为当前字符的子节点,并将当前结点设置为新创建的节点。
3).将当前字符设置为串中的下个字符,若当前字符为0,则结束;否则转2.
3.查找字符串
搜索过程与插入操作类似,当字符找不到匹配时返回假;若全部字符都存在匹配,判断最终停留的节点是否为树叶,若是,则返回真,否则返回假。
4.删除字符串
首先查找该字符串,边查询边将经过的节点压栈,若找不到,则返回假;否则依次判断栈顶节点是否为树叶,若是则删除该节点,否则返回真。
#include <iostream> using namespace std; /* trie的节点类型 */ template <int Size> //Size为字符表的大小 struct trie_node { bool terminable; //当前节点是否可以作为字符串的结尾 int node; //子节点的个数 trie_node *child[Size]; //指向子节点指针 /* 构造函数 */ trie_node() : terminable(false), node(0) { memset(child, 0, sizeof(child)); } }; /* trie */ template <int Size, typename Index> //Size为字符表的大小,Index为字符表的哈希函数 class trie { public: /* 定义类型别名 */ typedef trie_node<Size> node_type; typedef trie_node<Size>* link_type; /* 构造函数 */ trie(Index i = Index()) : index(i){ } /* 析构函数 */ ~trie() { clear(); } /* 清空 */ void clear() { clear_node(root); for (int i = 0; i < Size; ++i) root.child[i] = 0; } /* 插入字符串 */ template <typename Iterator> void insert(Iterator begin, Iterator end) { link_type cur = &root; //当前节点设置为根节点 for (; begin != end; ++begin) { if (!cur->child[index[*begin]]) //若当前字符找不到匹配,则新建节点 { cur->child[index[*begin]] = new node_type; ++cur->node; //当前节点的子节点数加一 } cur = cur->child[index[*begin]]; //将当前节点设置为当前字符对应的子节点 } cur->terminable = true; //设置存放最后一个字符的节点的可终止标志为真 } /* 插入字符串,针对C风格字符串的重载版本 */ void insert(const char *str) { insert(str, str + strlen(str)); } /* 查找字符串,算法和插入类似 */ template <typename Iterator> bool find(Iterator begin, Iterator end) { link_type cur = &root; for (; begin != end; ++begin) { if (!cur->child[index[*begin]]) return false; cur = cur->child[index[*begin]]; } return cur->terminable; } /* 查找字符串,针对C风格字符串的重载版本 */ bool find(const char *str) { return find(str, str + strlen(str)); } /* 删除字符串 */ template <typename Iterator> bool erase(Iterator begin, Iterator end) { bool result; //用于存放搜索结果 erase_node(begin, end, root, result); return result; } /* 删除字符串,针对C风格字符串的重载版本 */ bool erase(char *str) { return erase(str, str + strlen(str)); } /* 按字典序遍历单词树 */ template <typename Functor> void traverse(Functor &execute = Functor()) { visit_node(root, execute); } private: /* 访问某结点及其子结点 */ template <typename Functor> void visit_node(node_type cur, Functor &execute) { execute(cur); for (int i = 0; i < Size; ++i) { if (cur.child[i] == 0) continue; visit_node(*cur.child[i], execute); } } /* 清除某个节点的所有子节点 */ void clear_node(node_type cur) { for (int i = 0; i < Size; ++i) { if (cur.child[i] == 0) continue; clear_node(*cur.child[i]); delete cur.child[i]; cur.child[i] = 0; if (--cur.node == 0) break; } } /* 边搜索边删除冗余节点,返回值用于向其父节点声明是否该删除该节点 */ template <typename Iterator> bool erase_node(Iterator begin, Iterator end, node_type &cur, bool &result) { if (begin == end) //当到达字符串结尾:递归的终止条件 { result = cur.terminable; //如果当前节点可以作为终止字符,那么结果为真 cur.terminable = false; //设置该节点为不可作为终止字符,即删除该字符串 return cur.node == 0; //若该节点为树叶,那么通知其父节点删除它 } //当无法匹配当前字符时,将结果设为假并返回假,即通知其父节点不要删除它 if (cur.child[index[*begin]] == 0) return result = false; //判断是否应该删除该子节点 else if (erase_node((++begin)--, end, *(cur.child[index[*begin]]), result)) { delete cur.child[index[*begin]]; //删除该子节点 cur.child[index[*begin]] = 0; //子节点数减一 //若当前节点为树叶,那么通知其父节点删除它 if (--cur.node == 0 && cur.terminable == false) return true; } return false; //其他情况都返回假 } /* 根节点 */ node_type root; /* 将字符转换为索引的转换表或函数对象 */ Index index; }; //index function object class IndexClass { public: int operator[](const char key) { return key % 26; } }; int main() { trie<26,IndexClass> t; t.insert("tree"); t.insert("tea"); t.insert("A"); t.insert("ABC"); if(t.find("tree")) cout<<"find tree"<<endl; else cout<<"not find tree"<<endl; if(t.find("tre")) cout<<"find tre"<<endl; else cout<<"not find tre"<<endl; if(t.erase("tree")) cout<<"delete tree"<<endl; else cout<<"not find tree"<<endl; if(t.find("tree")) cout<<"find tree"<<endl; else cout<<"not find tree"<<endl; return 0; }