数据结构_trie

基本实现 存储 查找

import com.sun.org.apache.xpath.internal.operations.Bool;

import java.util.TreeMap;

public class MyTrie {
    private class Node{
        //该节点是不是一个单词的结尾
        public boolean isEnd;
        //有多少个孩子 并指向孩子节点
        public TreeMap<Character,Node> next;
        //空参构造
        public Node(){
            this.isEnd = false;
        }
        //用户指定
        public Node(boolean isEnd){
            this.isEnd = isEnd;
            next = new TreeMap<>();
        }
    }
    //树由节点构成
    private Node root;
    private int size;
    //构造参数
    public MyTrie(){
        root = new Node();
        size=0;
    }
    //获得单词数
    public int getSize(){
        return size;
    }
    //向树中添加单词
    public void add(String word){
        //将String 每一个字母 向树种添加 没有该节点的时候 创建 直到字符串遍历完毕
        Node cur = root;
        for(int i =0;i<word.length();i++){
            char w = word.charAt(i);
            //查看子节点是否包含此字母
            if(!cur.next.containsKey(w))
                //创建子节点
                cur.next.put(w,new Node());
            //转到子节点
            cur = cur.next.get(w);
        }
        //遍历结束 当前cur停在叶子节点上 判断此叶子节点 是否已经有 结尾标记 有的话证明 添加了重复字符串 无需改变 没有标记的话 需要标记
        if(!cur.isEnd){
            //新单词 且添加成功 size++
            cur.isEnd=true;
            size++;
        }
    }
    //是否包含某单词
    public Boolean contains(String word){
        Node cur = root;
        for(int i = 0;i<word.length();i++){
            if(cur.next.get(word.charAt(i))==null)
                return false;
            cur = cur.next.get(word.charAt(i));
        }
        return cur.isEnd;
    }
}