Tire树(字典树、前缀树)的简介和构造

最新推荐文章于 2024-08-04 16:58:42 发布

MaloryVer9

最新推荐文章于 2024-08-04 16:58:42 发布

阅读量846

点赞数

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/MaloryVer9/article/details/78358440

版权

数据结构专栏收录该内容

3 篇文章 0 订阅

订阅专栏

先留着填个坑，免得哪天忘记写了。。。
先把代码写下，注释以后再补，现在在等offer心焦中

namespace CjpSTL
{
    class Trie
    {
        struct Node;
        using NodePtr = Node*;

        struct Node
        {
            unordered_map<char, NodePtr> dict;

            char ch;
            bool hasVal;
            int count;
            Node(char _ch) :ch(_ch), dict(), count(1), hasVal(false) {}

        };

    public:
        Trie();

        void add(const string& str) const
        {
            auto cur = root;
            for (auto ch : str)
            {
                auto des = cur->dict.find(ch);
                if (des != cur->dict.end())
                {
                    ++cur->dict[ch]->count;
                    cur = cur->dict[ch];
                }
                else
                {
                    cur->dict[ch] = new Node(ch);
                    cur = cur->dict[ch];
                }

            }
            cur->hasVal = true;
        }
        bool find(const string& str)const
        {
            auto cur = root;
            for (auto ch : str)
            {
                auto des = cur->dict.find(ch);
                if (des != cur->dict.end())
                {

                    cur = cur->dict[ch];
                    ++cur->count;
                }
                else
                {
                    return false;
                }

            }
            return cur->hasVal;

        }
        void erase(const string& str)const
        {
            auto cur = root;
            auto pre = root;
            for (auto ch : str)
            {
                auto des = cur->dict.find(ch);
                if (des != cur->dict.end())
                {

                    pre = cur;
                    cur = cur->dict[ch];
                    --cur->count;
                }
                else
                {
                    break;
                }

            }
            cur->hasVal = false;


        }

        ~Trie();
    private:
        NodePtr root = new Node('\0');

    };

}