数据结构之字典树

Trie又被称为前缀树、字典树，所以当然是一棵树。上面这棵Trie树包含的字符串集合是{in, inn, int, tea, ten, to}。每个节点的编号是我们为了描述方便加上去的。树中的每一条边上都标识有一个字符。这些字符可以是任意一个字符集中的字符。比如对于都是小写字母的字符串，字符集就是’a’-‘z’；对于都是数字的字符串，字符集就是’0’-‘9’；对于二进制字符串，字符集就是0和1。

比如上图中3号节点对应的路径0123上的字符串是inn，8号节点对应的路径0568上的字符串是ten。终结点与集合中的字符串是一一对应的。

 

 

 

具体来说，Trie一般支持两个操作：
1.       Trie.insert(W)：第一个操作是插入操作，就是将一个字符串W加入到集合中。
2.       Trie.search(S)：第二个操作是查询操作，就是查询一个字符串S是不是在集合中

#include"字典树Trie.h"
#include<iostream>
using namespace std;

Trie* Trie::create_Trie() {
    Trie *root = (Trie*)malloc(sizeof(Trie));
    root->count = 0;
    for (int i = 0; i < MAX; i++) {
        root->next[i] = NULL;
    }
    return root;
}

void Trie::insert_word(Trie *root, char *str) {
    int len = strlen(str);
    Trie *tmp = root;
    for (int i = 0; i < len; i++) {
        int index = str[i] - 'a';
        if (root->next[index] == NULL) {
            Trie *tmp = new Trie();
            root->next[index] = tmp;
            root = root->next[index];
        }
        else {
            root = root->next[index];
            root->count += 1;
        }
    }
}
int Trie::count_Trie(Trie *root, char *str) {
    int len = strlen(str);
    for (int i = 0; i < len; i++) {
        int index = str[i] - 'a';
        if (!root->next[index])return 0;
        else
            root = root->next[index];
    }
    return root->count;
}

void Trie::destory_Trie(Trie *root) {
    for (int i = 0; i < MAX; i++) {
        if (root->next[i])destory_Trie(root->next[i]);
    }
    free(root);
}