二叉树算法之字典树（Trie）详细解读

字典树（Trie，也称前缀树或单词查找树）是一种用于快速查找字符串的数据结构，主要应用于字符串集合的高效存储和查找。字典树特别适合处理具有相同前缀的大量字符串集合，比如单词自动补全、拼写检查等场景。

1. 字典树的定义

字典树是一棵多叉树，通常用于存储字符串。树的每个节点代表一个字符，从根节点到某个节点的路径组成一个字符串。字典树的核心思想是利用字符串的公共前缀来减少存储空间和查找时间。

字典树具有以下几个特性：

根节点不包含字符，除根节点外每个节点只包含一个字符。
从根节点到某个节点的路径表示一个字符串的前缀。
每个节点的所有子节点包含的字符都不同。
每个叶子节点代表一个字符串的结束。

2. 字典树的结构

字典树由节点构成，每个节点存储以下信息：

字符：该节点表示的字符。
子节点的引用：指向其子节点的引用（即后续字符）。
结束标志：表示从根到该节点的路径是否构成了一个完整的字符串。

每个节点的子节点可以是26个字母（如果是只处理英文小写字母的话），但在实际实现中可以根据需求调整字符集的大小。

3. 字典树的基本操作

字典树支持以下基本操作：

插入（Insert）：将一个字符串插入到字典树中。
查找（Search）：判断字典树中是否存在某个字符串。
前缀查找（StartsWith）：判断字典树中是否存在某个前缀。

3.1 插入操作

插入操作的目的是将一个字符串插入到字典树中。如果该字符串的前缀已经存在于树中，则可以共用已有的节点，否则需要创建新的节点。

步骤：

从根节点开始，逐个检查字符串中的字符。
对于当前字符，如果在当前节点的子节点中找不到对应的字符节点，则创建一个新的节点。
移动到该字符的节点，继续处理下一个字符。
重复该过程，直到字符串的所有字符都处理完毕。
标记最后一个字符的节点为字符串的结束节点。

时间复杂度：插入操作的时间复杂度为 O(L)，其中 L 是要插入的字符串的长度。

3.2 查找操作

查找操作的目的是判断某个字符串是否存在于字典树中。该过程类似于插入操作。

步骤：

从根节点开始，逐个检查字符串中的字符。
对于每个字符，检查当前节点的子节点中是否存在对应的字符节点。如果不存在，则该字符串不在树中。
如果所有字符都找到，并且最后一个字符节点标记为字符串结束，则说明该字符串存在。

时间复杂度：查找操作的时间复杂度为 O(L)，其中 L 是要查找的字符串的长度。

3.3 前缀查找操作

前缀查找操作的目的是判断是否存在以某个字符串作为前缀的单词。

步骤：

从根节点开始，逐个检查前缀字符串中的字符。
对于每个字符，检查当前节点的子节点中是否存在对应的字符节点。如果不存在，则该前缀不在树中。
如果前缀中的所有字符都找到，说明字典树中存在以该前缀开头的单词。

时间复杂度：前缀查找操作的时间复杂度为 O(L)，其中 L 是前缀的长度。

4. 字典树的实现

下面是一个字典树的Java实现，支持插入、查找和前缀查找操作：

class TrieNode {// 每个节点包含的字符TrieNode[] children = new TrieNode[26];  // 假设只处理小写字母boolean isEndOfWord;  // 该节点是否为某个单词的结束public TrieNode() {isEndOfWord = false;for (int i = 0; i < 26; i++) {children[i] = null;}}
}class Trie {private TrieNode root;public Trie() {root = new TrieNode();}// 插入一个字符串public void insert(String word) {TrieNode node = root;for (int i = 0; i < word.length(); i++) {int index = word.charAt(i) - 'a';if (node.children[index] == null) {node.children[index] = new TrieNode();}node = node.children[index];}node.isEndOfWord = true;  // 标记字符串结束}// 查找一个字符串public boolean search(String word) {TrieNode node = root;for (int i = 0; i < word.length(); i++) {int index = word.charAt(i) - 'a';if (node.children[index] == null) {return false;  // 字符不存在}node = node.children[index];}return node != null && node.isEndOfWord;  // 判断是否为字符串结束}// 判断是否存在以某个前缀开头的字符串public boolean startsWith(String prefix) {TrieNode node = root;for (int i = 0; i < prefix.length(); i++) {int index = prefix.charAt(i) - 'a';if (node.children[index] == null) {return false;  // 前缀不存在}node = node.children[index];}return true;  // 前缀存在}
}public class Main {public static void main(String[] args) {Trie trie = new Trie();trie.insert("apple");System.out.println(trie.search("apple"));   // 输出 trueSystem.out.println(trie.search("app"));     // 输出 falseSystem.out.println(trie.startsWith("app")); // 输出 truetrie.insert("app");System.out.println(trie.search("app"));     // 输出 true}
}