空间优化的字典树

(同步个人博客 http://sxysxy.org/blogs/29 到csdn)

字典树空间优化

如果字典树要储存的字符串的字符集比较大，(比如全部的字符)，甚至可能有多字节字符。这是我们给每个节点256个子节点吗? 用平衡树(或map <char, int>)牺牲时间来换取空间吗？

不，可以进行这样的优化: 我们考虑把一个8位的char，拆开变成2个4位的数据，依次加入字典树。显然4位最大值也只是2^4-1=15，每个节点只需要16个子节点。 当然这样做相当于字符串长度len *= 2，时间 *= 2。但是如果原先需要的空间为m，这样优化后的空间只需要2*sqrt(m)，大幅度减小了空间占用。而时间并没增大很多。

下面给出一份实现代码

#ifndef TRIE_H
#define TRIE_H

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
#include "basedef.h"

#define NEXT_SIZE 16
typedef struct trieNodeType
{
    bool isFinal;
    struct trieNodeType *(*next);
}trieNode, *pTrieNode;

typedef struct trieType
{
    trieNode *root;
}trie;

void initTrie(trie *t)
{
    t -> root = (trieNode *)malloc(sizeof(trieNode));
    memset(t -> root, 0, sizeof(trieNode));
}
trieNode *appendChar(trieNode *d, int t)
{
    if(!d -> next)
    {
        d -> next = (pTrieNode *)malloc(sizeof(pTrieNode) * NEXT_SIZE);
        memset(d -> next, 0, sizeof(pTrieNode) * NEXT_SIZE);
    }
    if(!d -> next[t])
    {
        d -> next[t] = (pTrieNode)malloc(sizeof(trieNode));
        memset(d -> next[t], 0, sizeof(trieNode));
    }
    return d -> next[t];
}
void insertWord(trie *t, char *w, int len)
{
    trieNode *fw = t -> root;
    for(int i = 0; i < len; i++)
    {
        unsigned char c = w[i];
        unsigned hw = (c & (unsigned char)0xf0)>>4;   //高四位
        unsigned lw = c & (unsigned char)0x0f;        //低四位
        fw = appendChar(fw, hw);
        fw = appendChar(fw, lw);
    }
    fw -> isFinal = true;
}
bool existWord(trie *t, char *w, int len)
{
    trieNode *fw = t -> root;
    for(int i = 0; i < len; i++)
    {
        if(!fw -> next)
            return false;
        unsigned char c = w[i];
        unsigned hw = (c & (unsigned char)0xf0)>>4;
        unsigned lw = c & (unsigned char)0x0f;
        if(fw -> next[hw])
            fw = fw -> next[hw];
        else return false;
        if(fw -> next[lw])
            fw = fw -> next[lw];
        else return false;
        if(!fw)return false;
    }
    return fw -> isFinal;
}
void destroyTrieNodes(pTrieNode d)
{
    if(!d)return;
    if(!d -> next)return;
    for(int i = 0; i < NEXT_SIZE; i++)
        if(d -> next[i])destroyTrieNodes(d -> next[i]);
    for(int i = 0; i < NEXT_SIZE; i++)
        if(d -> next[i])free(d -> next[i]);
    free(d -> next);
}
void destroyTrie(trie *t)
{
    destroyTrieNodes(t -> root);
}

#endif