Trie(前缀树)

Trie的定义

字典树，英文名 trie。顾名思义，就是一个像字典一样的树。
Trie，又称前缀树或字典树，是一棵有根树，其每个节点包含以下字段：

1.指向子节点的指针数组children。children数组长度为 26，即小写英文字母的数量。此时 children[0] 对应小写字母 a，children[1] 对应小写字母 b，…，children[25] 对应小写字母 z。当然如果要存储更多的字符如数字0-9，或者汉字我们可以用哈希标来存储：如
HashMap<Integer, String> children = new HashMap<Integer, String>();
2.布尔字段 isEnd，表示该节点是否为字符串的结尾。

Tire结构图

实战题目

请你实现 Trie 类：

• Trie() 初始化前缀树对象。
• void insert(String word) 向前缀树中插入字符串 word 。
• boolean search(String word) 如果字符串 word 在前缀树中，返回 true（即，在检索之前已经插入）；否则，返回 false 。
• boolean startsWith(String prefix) 如果之前已经插入的字符串 word 的前缀之一为 prefix ，返回 true ；否则，返回 false 。

示例

输入

[“Trie”, “insert”, “search”, “search”, “startsWith”, “insert”, “search”]
[[], [“apple”], [“apple”], [“app”], [“app”], [“app”], [“app”]]
输出
[null, null, true, false, true, null, true]

解释

• Trie trie = new Trie();
• trie.insert(“apple”);
• trie.search(“apple”); // 返回 True
• trie.search(“app”); // 返回 False
• trie.startsWith(“app”); // 返回 True
• trie.insert(“app”);
• trie.search(“app”); // 返回 True

代码

class Trie {
    private Trie[] children;
    private int R = 26;
    private boolean isEnd;
    public Trie() {//初始化前缀树对象
        children = new Trie[R];
        isEnd = false;
    }

    public void insert(String word) {
        Trie node = this;
        for(int i = 0; i < word.length(); i++){
            char ch = word.charAt(i);
            int index = ch - 'a';//相对于首字母a的位置
            if(node.children[index] == null){
                node.children[index] = new Trie();
            }
            node = node.children[index];
        }
        node.isEnd = true;
    }
    
    public boolean search(String word) {
        Trie node = searchPrefix(word);
        return node != null && node.isEnd;
    }
    
    public boolean startsWith(String prefix) {
        return searchPrefix(prefix) != null;
    }

    public Trie searchPrefix(String prefix){
        Trie node = this;
        for(int i = 0; i < prefix.length(); i++){
            char ch = prefix.charAt(i);
            int index = ch - 'a';
            if(node.children[index] == null){
                return null;
            }
            node = node.children[index];
        }
        return node;        
    }
}