详解Java前缀树Trie的原理及代码实现

class Trie {
    /**
* 经过该节点的字符串的下层节点
*/
    Map<Character, Trie> next;
    /**
* 该节点是否是一个字符串的结束
*/
    boolean isEnd;

    public Trie() {
        this.next = new HashMap<>();
        this.isEnd = false;
    }

}

public void insert(String word) {
    //初始默认为根节点，根节点不包含任何字符
    Trie cur = this;
    //遍历该字符串的字符数组
    for (char c : word.toCharArray()) {
        //如果该节点的下层不包含此字符，那么加入一个新节点进去
        if (!cur.next.containsKey(c)) {
            cur.next.put(c, new Trie());
        }
        //查找下一层节点
        cur = cur.next.get(c);
    }
    //遍历字符串完毕，最后的节点isEnd置为true，表示一个字符串的结束
    cur.isEnd = true;
}

/**
 * 查找字符串
 */
public boolean search(String word) {
    Trie end = searchPrefix(word);
    return end != null && end.isEnd;
}

/**
 * 匹配前缀
 */
public boolean startsWith(String prefix) {
    return searchPrefix(prefix) != null;
}

private Trie searchPrefix(String prefix) {
    //初始默认为根节点，根节点不包含任何字符
    Trie cur = this;
    //遍历该字符串的字符数组
    for (char c : prefix.toCharArray()) {
        //如果该节点的下层不包含此字符，那么直接返回null
        if (!cur.next.containsKey(c)) {
            return null;
        }
        //查找下一层节点
        cur = cur.next.get(c);
    }
    return cur;
}

class Trie {
    /**
     * 经过该节点的字符串的下层节点
     */
    Map<Character, Trie> next;
    /**
     * 该节点是否是一个字符串的结束
     */
    boolean isEnd;

    public Trie() {
        this.next = new HashMap<>();
        this.isEnd = false;
    }

    public void insert(String word) {
        //初始默认为根节点，根节点不包含任何字符
        Trie cur = this;
        //遍历该字符串的字符数组
        for (char c : word.toCharArray()) {
            //如果该节点的下层不包含此字符，那么加入一个新节点进去
            if (!cur.next.containsKey(c)) {
                cur.next.put(c, new Trie());
            }
            //查找下一层节点
            cur = cur.next.get(c);
        }
        //遍历字符串完毕，最后的节点isEnd置为true，表示一个字符串的结束
        cur.isEnd = true;
    }

    /**
     * 查找字符串
     */
    public boolean search(String word) {
        Trie end = searchPrefix(word);
        return end != null && end.isEnd;
    }

    /**
     * 匹配前缀
     */
    public boolean startsWith(String prefix) {
        return searchPrefix(prefix) != null;
    }

    private Trie searchPrefix(String prefix) {
        //初始默认为根节点，根节点不包含任何字符
        Trie cur = this;
        //遍历该字符串的字符数组
        for (char c : prefix.toCharArray()) {
            //如果该节点的下层不包含此字符，那么直接返回null
            if (!cur.next.containsKey(c)) {
                return null;
            }
            //查找下一层节点
            cur = cur.next.get(c);
        }
        return cur;
    }

    public static void main(String[] args) {
        Trie trie = new Trie();
        trie.insert("你是xxl吗？");
    }
}