Trie模板

最新推荐文章于 2024-03-19 13:35:45 发布

原创最新推荐文章于 2024-03-19 13:35:45 发布 · 269 阅读

0 ·

CC 4.0 BY-SA版权

模板同时被 2 个专栏收录

3 篇文章

订阅专栏

Trie

1 篇文章

订阅专栏

本文详细介绍了两种不同的Trie树实现方式：一种是基于字符数组的普通Trie树，另一种则是采用左儿子右兄弟结构的Trie树。通过具体代码展示了如何进行字符串插入和查找操作。

普通Trie

const int ALPHABETS=26;
struct TrieNode {
    TrieNode *c[ALPHABETS];
    bool f;

    TrieNode():f(false) {
        memset(c,0,sizeof c);
    }

    ~TrieNode() {
        for(int i=0;i<ALPHABETS;i++)
            if(c[i])
                delete c[i];
    }

    void insert(const char *key) {
        if((*key)==0) {
            f=true;
        } else {
            int Next=((*key)-'A');
            if(c[Next]==NULL) {
                c[Next]=new TrieNode();
            }
            c[Next]->insert(key+1);
        }
    }

    TrieNode *find(const char *key) {
        if((*key)==0) {
            if(f)
                return this;
            else return NULL;
        }
        int Next=((*key)-'A');
        if(c[Next]==NULL) return NULL;
        return c[Next]->find(key+1);
    }
};

左儿子右兄弟

struct TrieNode {
    bool f;
    TrieNode *lc,*rb;
    char ch;

    TrieNode():f(false),lc(0),rb(0),ch(0) {}

    ~TrieNode() {
        delete lc;
        delete rb;
    }

    void insert(const char *key) {
        if((*key)==0) {
            f=true;
        } else {
            TrieNode *pos=lc,*lastpos=0;
            while(pos!=NULL) {
                if(pos->ch==(*key)) break;
                lastpos=pos;
                pos=pos->rb;
            }
            if(pos==NULL) {
                pos=new TrieNode();
                pos->ch=(*key);
                if(lastpos!=NULL)
                    lastpos->rb=pos;
            }
            if(lc==NULL) lc=pos;
            pos->insert(key+1);
        }
    }

    TrieNode *find(const char *key) {
        if((*key)==0) {
            if(this->f)
                return this;
            else return NULL;
        }
        TrieNode *pos=lc;
        while(pos!=NULL && pos->ch!=(*key))
            pos=pos->rb;
        if(pos==NULL) return NULL;
        else return pos->find(key+1);
    }
};