【LeetCode】Day136-实现 Trie (前缀树)

题目

208. 实现 Trie (前缀树)【中等】

题解

Trie 是一颗非典型的多叉树模型，一棵包含三个单词 “sea”,“sells”,“she” 的 Trie 为如下结构：

忽略空链接后可以理解为下面这棵树：

Trie 树中每个节点包含以下字段：

1. 指向子节点的指针数组 children。对于本题而言，数组长度为 26，即小写英文字母的数量。此时 children[0] 对应小写字母 a，children[1] 对应小写字母 b，…，children[25] 对应小写字母 z。
2. 布尔字段 isEnd，表示该节点是否为字符串的结尾。

class Trie {
    private Trie[] children;
    private boolean isEnd;//是否是单词结尾

    public Trie() {
        children=new Trie[26];
    }
    
    public void insert(String word) {
        Trie node=this;
        for(int i=0;i<word.length();i++){
            int index=word.charAt(i)-'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=find(word);
        return node!=null&&node.isEnd;
    }
    
    public boolean startsWith(String prefix) {
        return find(prefix)!=null;
    }
    //找字符串str的末尾结点
    public Trie find(String str){
        Trie node=this;
        for(int i=0;i<str.length();i++){
            int index=str.charAt(i)-'a';
            //没找到str
            if(node.children[index]==null)
                return null;
            node=node.children[index];
        }
        return node;
    }
}

时间复杂度：$O(|S|)$，S为每次插入或者删除的字符串长度

空间复杂度：$O(26\ast |T|)$，T为插入字符串的长度之和