leetcode > 208. 实现 Trie (前缀树) #字典树 #难度中等 #Java

leetcode > 208. 实现 Trie (前缀树) #字典树 #难度中等 #Java

审题

题目链接(https://leetcode-cn.com/problems/implement-trie-prefix-tree/)

说明：由题目描述和右边代码示例可以看出，我们要实现一个Trie树，包含一个无参构造方法，三个功能，分别是插入一个字符串、查询一个字符串、判断trie树中是否存在一个字符前缀。其中需要注意的是search(String)和startsWith(String)方法的区别。

分析

Trie树也称为字典树，它是一种专门处理字符串匹配的数据结构，用来解决在一组字符串集合中快速查找某个字符串的问题。那么Trie树这种数据结构是什么样子的呢？首先，我们假如有6个字符串：how， hi， her， hello， so， see需要保存在Trie树中，这时Trie树如下图所示：

如上图所示Trie树是一种多叉树，如果只保存英文的26个字母（不区分大小写），那么Trie树每个节点最多有26个分叉，要是按照二叉树的数据结构每个节点保存了左右子节点的指针，难以想象Trie树每个节点要保存26个子节点的指针？假如我们将这26个节点的指针按照数组保存，那么就是每个节点保存了一个26个子节点的数组。借助映射的思想，我们将字符和数组下标做映射，我们在数组中下标为0的位置，存储指向子节点a的指针，下标为1的位置存储指向子节点b的指针，以此类推，下标为25的位置，存储的是指向的子节点z的指针。如果某个字符的子节点不存在，我们就在对应的下标的位置存储null。

class TrieNode {
	char data;
	TrieNode children[26];
}

那么和上面一样，我们只保存6个字符串：how， hi， her， hello， so， see，Trie树的第一和第二层（排除root节点）就是下面这样：

代码如下

class Trie {

    private TrieNode root;

    public Trie() {
        // 存储无意义字符
        root = new TrieNode('/');
    }

    // 往Trie树中插入一个字符串
    public void insert(String word) {
        char[] text = word.toCharArray();
        TrieNode p = root;
        for (int i = 0; i < text.length; ++i) {
            int index = text[i] - 'a';
            if (p.children[index] == null) {
                TrieNode newNode = new TrieNode(text[i]);
                p.children[index] = newNode;
            }
            p = p.children[index];
        }
        p.isEndingChar = true;
    }

    // 在Trie树中查找一个字符串
    public boolean search(String word) {
        char[] pattern = word.toCharArray();
        TrieNode p = root;
        for (int i = 0; i < pattern.length; ++i) {
            int index = pattern[i] - 'a';
            if (p.children[index] == null) {
                // 不存在pattern
                return false;
            }
            p = p.children[index];
        }
        if (p.isEndingChar == false) {
            // 不能完全匹配，只是前缀
            return false;
        } else {
            // 找到pattern
            return true;
        }
    }

    public boolean startsWith(String prefix) {
        char[] pattern = prefix.toCharArray();
        TrieNode p = root;
        for (int i = 0; i < pattern.length; ++i) {
            int index = pattern[i] - 'a';
            if (p.children[index] == null) {
                // 不存在pattern
                return false;
            }
            p = p.children[index];
        }
        return true;
    }

    public class TrieNode {
        public char data;
        public TrieNode[] children = new TrieNode[26];
        public boolean isEndingChar = false;

        public TrieNode(char data) {
            this.data = data;
        }
    }
}

参考

王争《数据结构与算法之美》 -极客时间