字典树

1.字典树概论

字典树（trie），又称前缀树。是查找树的一种。主要用于大量字符串的存储，查找。搜索引擎通常利用字典树完成用户的搜索提示。

与通常的查找树不同，字典树的节点并不存储键(key), 节点的键有节点在树中的位置决定。从树的根开始到某一节点开始遍历可以得到该节点的键。每个节点的键的信息存储在节点间的边中。所有节点的后继的前缀都存储在该节点的键中。根结点与空字符串关联。如下图所示，该字典树存储的字符串有 to, tea, ted, ten, inn。虽然图中每个节点都标示有一个键，但实际上每个节点只有26个指针（假设字符串不区分大小写）。

2.代码实例

字典树的每个节点包含3个字段，exist标示该结点是否为一个有效的单词。next是26个指向其他节点的指针

typedef struct trie_node {
        int exist; //if there is a word, 0 no word, 1 opposite
        struct trie_node *next[ALPHT];
}trie_t;

字典树主要有查找和插入两个操作

trie_t *creat_node()
{
        trie_t *node = (trie_t *)malloc(sizeof(trie_t));
        node->cnt = 0;
        node->exist = 0;
        int i;
        for (i = 0; i < ALPHT; i++)
                node->next[i] = NULL;
        return node;
}
int insert_word(trie_t *root, char *key)
{
        char *strp = key;
        trie_t *p = root;
        int i = 0;

        while(p->next[strp[i]-'a'] != NULL) {
                p = p->next[strp[i]-'a'];
                i++;
        }
        while(strp[i] != '\0') {
                trie_t *node = creat_node();
                node->cnt++;
                p->next[strp[i]-'a'] = node;
                p = node;
                i++;
        }

        p->exist = 1;

        return p->cnt;
}
int trie_search(trie_t *root, char *key)
{
        trie_t *p = root;
        char *strp = key;
        int i = 0, flag = 0;

        while(p->next[strp[i]-'a'] != NULL) {
                p = p->next[strp[i]-'a'];
                i++; 
        }
        if (strlen(strp) == i) {
                flag = 1;
                printf("find word %s\n", strp);
        }
        else flag = 0;
        return flag;
}
int main(void) 
{
        char words[][MAXLINE] = {"honey", "uber", "wisdom", "hit", "fool"};
        int i;
        //create the root node
        trie_t *root = creat_node();
        for (i = 0; i < 5; i++)
                insert_word(root, words[i]);

        int flag;
        flag = trie_search(root, "honey");
        printf("%d\n", flag);
        return 0;
}

3.复杂度分析

字典树查找的时间复杂度和字符串长度M有关，所以其复杂度为O(M)。字典树的缺陷是占用空间过大。