【hot100-java】【实现 Trie (前缀树)】

R8-图论篇

class Trie {

    //记录该字母的下一位所有可能的字母坐标
    private Trie[] children;
    //该字母是否为最后一个字母
    private boolean isEnd;

    public Trie() {
          //初始化26个字母
          children=new Trie[26];
          //默认为不是最后一个字母
          isEnd=false;
    }
    
    public void insert(String word) {
         //得到字典树根节点
         Trie node=this;
         //去遍历待插入单词的字符集合
         for (char c:word.toCharArray()){
             //得到该字符再数组中的坐标
             int index=c-'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);
        //只有当该单词再字典树中存在并且最后一个字母结点为最后一个字母，才返回true
        return node!=null && node.isEnd;
    }
    
    public boolean startsWith(String prefix) {
          //只要前缀匹配存在于字典树中就返回true
          return searchPrefix(prefix)!=null;
    }
    //前缀搜索 还是全文搜索都是调用该方法，区别在于前缀搜索只要前缀匹配就返回true，全文搜索则要匹配到最后一个字母才返回true，所以这里返回的是最后一个字母节点
    public Trie searchPrefix(String word){
        //得到字典树根节点
        Trie node=this;
        //开始验证字符串在字典树是否存在
        for (char c:word.toCharArray()){
            //得到该字符在数组中的坐标
            int index=c-'a';
            //存在，就会移动到下一个字母结点
            if (node.children[index]!=null){
                node=node.children[index];
            }else{
                //如果没有搜索到
                return null;
            }
        }
        //到达最后一个字母结点
        return node;
    }
}

/**
 * Your Trie object will be instantiated and called as such:
 * Trie obj = new Trie();
 * obj.insert(word);
 * boolean param_2 = obj.search(word);
 * boolean param_3 = obj.startsWith(prefix);
 */

附 

cy

java语法不难，这题倒挺难的。