Trie(前缀树)的java实现

实现 Trie 类：

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

class Trie {

        Trie[] children;
        boolean isEnd;
        public Trie() {
            children=new Trie[26];
            isEnd=false;
        }

        public void insert(String word) {
            Trie node=this;
            for (int i = 0; i < word.length(); i++) {
                char ch=word.charAt(i);
                if(node.children[ch-'a']==null)
                    node.children[ch-'a']=new Trie();
                node=node.children[ch-'a'];
            }
            node.isEnd=true;
        }

        public boolean search(String word) {
            Trie node=isPre(word);
            return node!=null && node.isEnd;
        }

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

        public boolean startsWith(String prefix) {
            return isPre(prefix)!=null;
        }
}