Trie树的C++实现

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 objectclass 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;}

http://en.wikipedia.org/wiki/Trie