字典树、前缀树 (Prefix Tree)详细解读

最新推荐文章于 2025-03-09 10:32:18 发布

Katie。

最新推荐文章于 2025-03-09 10:32:18 发布

阅读量1.1k

点赞数 21

分类专栏：数据结构文章标签： linux 运维服务器

本文链接：https://blog.csdn.net/m0_61840987/article/details/143717828

版权

数据结构专栏收录该内容

32 篇文章

订阅专栏

字典树 (Trie) / 前缀树 (Prefix Tree) 详解

字典树 (Trie)，又称为 前缀树 (Prefix Tree)，是一种树形数据结构，主要用于处理字符串集的数据。它特别适合用于 快速搜索 和 字符串处理 场景，如自动补全、拼写检查、前缀匹配等。Trie 的设计使得它能够高效地处理 字符串集合 中的公共前缀问题，从而节省存储空间和提高查询速度。

一、字典树 (Trie) 的基本结构

1. 结构定义

节点 (Node)：每个节点表示一个字符。一个节点可以有多个子节点。
根节点 (Root)：Trie 的根节点通常为空字符 ' '，即没有实际字符意义。
边 (Edge)：节点之间的连接表示字符的顺序。
终止标志 (End of Word)：用于标记一个单词的结束。通常通过一个布尔变量 isEnd 来表示。

2. 结构示例

假设我们插入以下单词："cat"、"cap"、"bat"，它们的 Trie 结构如下：

         (root)
        /       \
       c         b
     /   \        \
    a     a        a
   /       \        \
  t         p        t
(isEnd)   (isEnd)  (isEnd)

共有三个单词：cat, cap, bat。
公共前缀得到了有效存储节省，如 c -> a 只存储一次。

二、Trie 的基本操作

1. 插入操作 (Insert)

插入一个单词 word 时，逐个字符插入到 Trie 中。如果路径上没有对应的字符节点，则创建新的节点；否则沿着已有路径继续，直到插入完整个单词，并将最后一个字符标记为单词结尾。

示例代码 (Java)：

class TrieNode {
    TrieNode[] children = new TrieNode[26]; // 假设仅限于小写字母
    boolean isEndOfWord = false;
}

class Trie {
    private TrieNode root;

    public Trie() {
        root = new TrieNode();
    }

    // 插入单词
    public void insert(String word) {
        TrieNode current = root;
        for (char ch : word.toCharArray()) {
            int index = ch - 'a';
            if (current.children[index] == null) {
                current.children[index] = new TrieNode();
            }
            current = current.children[index];
        }
        current.isEndOfWord = true;
    }
}

2. 搜索操作 (Search)

搜索一个单词时，从根节点开始沿着 Trie 的路径查找每个字符。如果找不到对应的路径，则表示单词不存在；如果能找到完整路径并且结尾节点标记为 isEndOfWord，则表示单词存在。

示例代码 (Java)：

// 查找单词
public boolean search(String word) {
    TrieNode current = root;
    for (char ch : word.toCharArray()) {
        int index = ch - 'a';
        if (current.children[index] == null) {
            return false;
        }
        current = current.children[index];
    }
    return current.isEndOfWord;
}

3. 前缀匹配 (StartsWith)

判断是否存在以给定前缀开头的单词。类似于 search() 操作，但不需要检查 isEndOfWord 标记。

示例代码 (Java)：

// 前缀匹配
public boolean startsWith(String prefix) {
    TrieNode current = root;
    for (char ch : prefix.toCharArray()) {
        int index = ch - 'a';
        if (current.children[index] == null) {
            return false;
        }
        current = current.children[index];
    }
    return true;
}