【Hot 100】208. 实现 Trie (前缀树)

已于 2025-05-13 10:39:56 修改
阅读量570 收藏 7
点赞数 26
分类专栏： 算法 文章标签： 开发语言 Hot100 leetcode c++ 数据结构 算法
于 2025-05-13 10:34:37 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/hhw_hhw/article/details/147917169
版权

目录

请添加图片描述

  • 🙋‍♂️ 作者：海码007
  • 📜 专栏：算法专栏
  • 💥 标题：【Hot 100】208. 实现 Trie (前缀树)
  • ❣️ 寄语：书到用时方恨少，事非经过不知难！

引言

实现 Trie (前缀树)

  • 🎈 题目链接：
  • 🎈 做题状态：

我的解题

首先需要理解前缀树的定义，前缀树是一颗多叉树，树根不存储字母。每一层可能存储26个不同的字母。然后每一个单词对应这个多叉树的一条路径，并且路径的结尾会标识是单词的结尾。

class Trie {
private:
    bool isEnd;
    Trie* next[26]; //指针数组，有26个小写字母

public:
    Trie() {
        isEnd = false;
        memset(next, 0, sizeof(next));
    }
    
    // 插入一个单词
    void insert(string word) {

        // node指向根节点并向下遍历
        Trie* node = this;
        for (char c : word)
        {
            // 判断当前这个字母是否在当前层存在，如果不存在则创建一个新的树。
            if (node->next[c-'a'] == nullptr)
            {
                node->next[c-'a'] = new Trie();
            }
            node = node->next[c-'a'];   // 继续往下遍历
        }

        // 遍历到末尾后，需要标识 end
        node->isEnd = true;
    }
    
    // 搜索当前单词是否存在，依次比较每个单词是否存在每一层中
    bool search(string word) {
        Trie* node = this;
        for (const char& c : word)
        {
            if (node->next[c-'a'] == nullptr)
            {
                return false;
            }
            node = node->next[c-'a'];
        }

        return node->isEnd; // 遍历到末尾后还需要判断是否是单词的结尾
    }
    
    // 判断这个前缀树是否包含 prefix 这个前缀
    bool startsWith(string prefix) {
        Trie* node = this;
        for (const char& c : prefix)
        {
            if (node->next[c-'a'] == nullptr)
            {
                return false;
            }
            node = node->next[c-'a'];
        }

        return true;
    }
};

/**
 * Your Trie object will be instantiated and called as such:
 * Trie* obj = new Trie();
 * obj->insert(word);
 * bool param_2 = obj->search(word);
 * bool param_3 = obj->startsWith(prefix);
 */

代码解析

该代码实现了一个基本的前缀树（Trie）结构，支持插入、搜索和前缀匹配功能。以下是对代码的详细分析和优化建议：

代码思路分析

  1. 数据结构设计：

    • isEnd：标记当前节点是否为单词的结尾。
    • next[26]：一个包含 26 个指针的数组，对应英文小写字母，用于构建字符到子节点的映射。

  2. 核心方法：

    • 插入 (insert): 从根节点开始，逐字符向下遍历。若字符对应的子节点不存在，则创建新节点。遍历完成后，标记最后一个节点为单词结尾。
    • 搜索 (search): 检查单词是否存在且最后一个节点被标记为结尾。
    • 前缀匹配 (startsWith): 仅检查路径是否存在，无需判断结尾标记。

  3. 正确性：

    • 逻辑正确，符合前缀树的标准实现。
    • 使用 memset 初始化指针数组，确保初始值为 nullptr。

优化建议

1. 内存泄漏问题

当前代码未释放动态分配的子节点，导致内存泄漏。需添加析构函数递归释放子节点：

~Trie() {
    for (int i = 0; i < 26; ++i) {
        delete next[i]; // 递归调用子节点的析构函数
    }
}
2. 使用智能指针优化内存管理

改用 std::unique_ptr 管理子节点，避免手动释放内存：

class Trie {
private:
    bool isEnd;
    std::array<std::unique_ptr<Trie>, 26> next; // 使用智能指针

public:
    Trie() : isEnd(false) {} // 无需 memset

    void insert(string word) {
        Trie* node = this;
        for (char c : word) {
            if (!node->next[c - 'a']) {
                node->next[c - 'a'] = std::make_unique<Trie>();
            }
            node = node->next[c - 'a'].get();
        }
        node->isEnd = true;
    }

    // search 和 startsWith 无需修改
};
3. 输入合法性校验（可选）

若需处理非法字符（如大写字母），可在插入/查询前添加校验：

void insert(string word) {
    for (char c : word) {
        if (c < 'a' || c > 'z') {
            throw std::invalid_argument("Invalid character");
        }
    }
    // 原插入逻辑
}
4. 其他优化
  • 空间优化：使用压缩 Trie 树（如 Ternary Search Tree）减少内存占用，但会增加实现复杂度。
  • 性能优化：当前时间复杂度为 O(L)（L 为字符串长度），已是最优，无需调整。

总结

代码逻辑正确，但存在内存泄漏问题。建议通过析构函数或智能指针优化内存管理。其他优化可根据实际需求选择。改进后的代码示例（使用智能指针）如下：

#include <memory>   // 用于智能指针 unique_ptr
#include <array>    // 用于固定大小的数组 array
#include <string>   // 用于字符串操作

class Trie {
private:
    // 标记当前节点是否为某个单词的结尾
    bool isEnd;
    
    // 使用智能指针管理子节点，避免内存泄漏
    // 数组大小为26，对应英文小写字母a-z
    std::array<std::unique_ptr<Trie>, 26> next;

public:
    // 构造函数：初始化 isEnd 为 false，表示初始时不是单词结尾
    // 智能指针数组 next 会自动初始化为 nullptr
    Trie() : isEnd(false) { }
    
    /**
     * 插入一个单词到 Trie 树中
     * @param word 待插入的单词
     */
    void insert(const std::string& word) {
        // 从根节点（this）开始遍历
        Trie* node = this;
        
        // 逐个字符处理
        for (char c : word) {
            // 计算字符对应的索引（a->0, b->1, ..., z->25）
            int idx = c - 'a';
            
            // 如果当前字符的子节点不存在，则创建新节点
            if (node->next[idx] == nullptr) {
                node->next[idx] = std::make_unique<Trie>();
            }
            
            // 移动到子节点继续处理
            node = node->next[idx].get();  // get() 获取裸指针
        }
        
        // 标记单词的最后一个字符节点为结尾
        node->isEnd = true;
    }
    
    /**
     * 搜索 Trie 树中是否存在某个单词
     * @param word 待搜索的单词
     * @return 如果单词存在且完整匹配（最后一个字符是结尾），返回 true；否则返回 false
     */
    bool search(const std::string& word) {
        // 从根节点开始遍历
        Trie* node = this;
        
        // 逐个字符检查
        for (const char& c : word) {
            int idx = c - 'a';
            
            // 如果当前字符的子节点不存在，说明单词不存在
            if (node->next[idx] == nullptr) {
                return false;
            }
            
            // 移动到子节点继续检查
            node = node->next[idx].get();
        }
        
        // 检查最后一个字符是否被标记为单词结尾
        return node->isEnd;
    }
    
    /**
     * 检查 Trie 树中是否存在某个前缀
     * @param prefix 待检查的前缀
     * @return 如果前缀存在（不要求是完整单词），返回 true；否则返回 false
     */
    bool startsWith(const std::string& prefix) {
        // 从根节点开始遍历
        Trie* node = this;
        
        // 逐个字符检查
        for (const char& c : prefix) {
            int idx = c - 'a';
            
            // 如果当前字符的子节点不存在，说明前缀不存在
            if (node->next[idx] == nullptr) {
                return false;
            }
            
            // 移动到子节点继续检查
            node = node->next[idx].get();
        }
        
        // 只要路径存在，无论是否是单词结尾，都返回 true
        return true;
    }
};
【NO.62】LeetCode HOT 100—208. 实现 Trie (前缀树)
wang_luwei的博客
03-10 990
208. 实现 Trie (前缀树)
Leetcode Hot100 第八题 208. 实现 Trie (前缀树)
weixin_45850570的博客
01-22 102
【代码】Leetcode Hot100 第八题 208. 实现 Trie (前缀树)
力扣HOT100 208. 实现Trie（前缀树）
qq_61504864的博客
04-28 367
【代码】力扣HOT100 208. 实现Trie（前缀树）
【LeetCode】hot100— 208. 实现 Trie (前缀树)
weixin_44382896的博客
03-24 1388
注：虽然每个 Trie 结点都有一个 长为 24 的 children数组，但每个数组中存储的是其子节点的结点值，例如 hello 中的 h 结点，仅有 children数组中 仅有。重复以上步骤，直到处理字符串的最后一个字符，然后将当前节点标记为字符串的结尾。若搜索到了前缀的末尾，就说明字典树中存在该前缀。我们从字典树的根开始，插入字符串。我们从字典树的根开始，查找前缀。重复以上步骤，直到返回空指针或搜索完前缀的最后一个字符。（发音类似 “try”）或者说。，表示该节点是否为字符串的结尾。
【Hot100】LeetCode—208. 实现 Trie (前缀树)
weixin_44382896的博客
08-27 501
前缀树的当前结点，存储的是下一个子节点的元素值。当前结点中的内容是一个哈希表，和 isEnd 字段。
208. 实现 Trie (前缀树)
s773364的专栏
04-14 246
类外定义节点，类内定义根节点。
【LeetCode】【HOT】208. 实现 Trie (前缀树)
2022年毕业的萌新程序猿，输出一下自己的学习内容~
01-20 251
【LeetCode】【HOT】208. 实现 Trie (前缀树)
Hot100【七十七】： 208. 实现 Trie (前缀树)
一只小星星
04-16 180
在写这道算法的时候，如果对这个数据结构比较熟悉，其实不是很难。
【力扣】[热题HOT100] 208. 实现Trie（前缀树）
金金金金金的博客
05-22 170
1、题目 Trie（发音类似 “try”）或者说 前缀树 是一种树形数据结构，用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景，例如自动补完和拼写检查。 请你实现 Trie 类： Trie() 初始化前缀树对象。 void insert(String word) 向前缀树中插入字符串 word 。 boolean search(String word) 如果字符串 word 在前缀树中，返回 true（即，在检索之前已经插入）；否则，返回 false 。 boolean start
力扣hot100--＞前缀和/前缀树/LRU缓存
Dong雨的博客
11-28 649
中等给你一个整数数组nums和一个整数k，请你统计并返回该数组中和为k的子数组的个数。子数组是数组中元素的连续非空序列。22public:// 获取输入数组的大小// 哈希表，用来存储前缀和的频次// 初始化哈希表，表示前缀和为0出现1次（这对从索引0开始的子数组非常重要）// 存储前缀和的数组（pre[i] 表示 nums[0] 到 nums[i-1] 的和）// 用于存储满足条件的子数组个数i < n;++i) {// 更新当前的前缀和。
LeetCode Hot 100 代码实现
weixin_45595437的博客
08-22 615
文章目录1. 两数之和（Easy，哈希）49. 字母异位词分组（Medium，哈希）128. 最长连续序列（Medium，哈希）283. 移动零（Easy，双指针）11. 盛最多水的容器（Medium，贪心（用双指针实现））15. 三数之和（Medium，双指针）42. 接雨水（Hard，贪心（用双指针实现））3. 无重复字符的最长子串（Medium，滑动窗）438. 找到字符串中所有字母异位词（Medium，滑动窗）560. 和为 K 的子数组（Medium，前缀和 + 哈希表的综合应用）239. 滑动窗
Leetcode HOT 100 (Java 版)
dwtianwen的博客
07-14 1247
一刷 Leetcode HOT 100 记录
鸿蒙系统使用ArkTS开发语言支持身份证阅读器、社保卡读卡器等调用二次开发SDK
广东东信智能科技有限公司的专栏
05-10 1143
this.tvResult = "读卡失败：" + ssCardInfor.result+"\n errMsg = "+ssCardInfor.errMsg.trim()+"有效期限："+ idInfo.signdate+" - "+ idInfo.validterm+"\n"+ "开始有效期限：" + ssCardInfor.signdate + "\n"+ "身份证号：" + ssCardInfor.idnumber + "\n"+ "性 别：" + ssCardInfor.sex + "\n"
力扣-94.二叉树的中序遍历
weixin_46878941的博客
05-12 171
小结：做简单题就是心情舒畅，注意单写一个函数就好了。给定一个二叉树的根节点 root ，返回 它的。
【Leetcode 每日一题】2094. 找出 3 位偶数
冠位优化
05-12 764
Leetcode 每日一题 - 找出 3 位偶数
LeetCode 513 找树左下角的值 LeetCode 112 路径总和 LeetCode106 从中序与后序遍历序列构造二叉树
weixin_59234335的博客
05-12 570
中序遍历 + 前序/后序遍历 可以 确定一个唯一的二叉树。必须有中序遍历的存在。而前序遍历+后序遍历无法确认。前序遍历：中左右后序遍历：左右中根据前序遍历和后序遍历的特性，我们只能唯一确定根节点，但无法从前序遍历和后序遍历划分出左右。例子如下：上面这两颗树的前序遍历都为[ 1, 2, 3] ，后序遍历为[3, 2, 1]
力扣-24.两两交换链表中的结点
weixin_46878941的博客
05-09 281
给你一个链表，两两交换其中相邻的节点，并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题（即，只能进行节点交换）。小结：后来才看到不能修改结点值，但是也不难。
力扣HOT100之二叉树：101. 对称二叉树
weixin_52151595的博客
05-12 170
这道题我本来想着挑战一下自己，尝试着用迭代的方法来做，然后就是用层序遍历，将每一层的元素收集到一个临时的一维向量中，然后再逐层判断每一层是否都是轴对称的，一旦发现某一层不是轴对称的，就直接。注意，再判断每一层是否为轴对称时，千万不能用栈来匹配（类似于括号匹配的思路），因为下面的情况用栈会判定为轴对称，但是实际上是非轴对称的。此外，当前层的下一层如果至少有一个非空节点，那么下一层的空节点也必须加入，否则判定也会出问题。的情况，就说明每一层都是轴对称的，此时直接。，当所有的层都遍历结束后，都没有出现。
力扣 旋转图像
最新发布
fancy166166的博客
05-12 231
创建一个新的数组，存储旋转后的数组，将数组的第一行存到新创的数组空间的最后一列。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值