zoukankan      html  css  js  c++  java
  • 10-3 Trie前缀树的查询【查询前缀树中是否包含指定的单词】

    import java.util.TreeMap;
    
    public class Trie {
    
        private class Node{
    
            public boolean isWord;
            public TreeMap<Character, Node> next;
    
            public Node(boolean isWord){
                this.isWord = isWord;
                next = new TreeMap<>();
            }
    
            public Node(){
                this(false);
            }
        }
    
        private Node root;
        private int size;
    
        public Trie(){
            root = new Node();
            size = 0;
        }
    
        // 获得Trie中存储的单词数量
        public int getSize(){
            return size;
        }
    
        // 向Trie中添加一个新的单词word
        public void add(String word){
    
            Node cur = root;
            for(int i = 0 ; i < word.length() ; i ++){
                char c = word.charAt(i);
                if(cur.next.get(c) == null)
                    cur.next.put(c, new Node());
                cur = cur.next.get(c);
            }
    
            if(!cur.isWord){
                cur.isWord = true;
                size ++;
            }
        }
    
        // 查询单词word是否在Trie中
        // 查询和添加的逻辑很相似,区别在于是否创建新节点,在添加方法中,如果孩子节点不存在,则创建新节点,在查询方法中,如果孩子节点不存在,直接返回false告诉调用者查找的单词不存在!
        public boolean contains(String word){
    
            Node cur = root;
            for(int i = 0 ; i < word.length() ; i ++){
                char c = word.charAt(i);
                if(cur.next.get(c) == null)
                    return false;
                cur = cur.next.get(c);
            }
            // 注意:这里不能直接返回true,因为一个前缀可能是另一个单词,
            // 比如:panda这个单词的前三个字母pan刚好是另一个单词,
            // 例如:Trie中存在panda,不存在pan,假如直接返回true,那么在查询pan这个单词时,也能查到,逻辑就是错误,所以应该返回cur.isWord
            return cur.isWord;
        }
    }
    

      

  • 相关阅读:
    Android ViewTreeObserver的常用技巧
    android 三种动画详解
    mysql binlog2sql闪回数据
    linux 系统时间与硬件时间与中国标准时间
    mysql 远程备份binlog日志
    关于ActivityThread类的深入学习
    android 全屏dialog dialogfragment
    mysql show slave 与 从库复制新的从库
    Android 软键盘的全面解析,让你不再怕控件被遮盖
    Android 隐藏输入法键盘
  • 原文地址:https://www.cnblogs.com/lpzh/p/12551777.html
Copyright © 2011-2022 走看看