Java实现前缀树(leetCode 208. 实现 Trie (前缀树))

文章目录

    • 什么是前缀树
    • 前缀树的作用
    • 使用HashMap实现前缀树

什么是前缀树

前缀树Trie Tree,是一种字典树,匹配树。
①根节点不包含字符,除根节点外的每一个子节点都包含一个字符。
②从根节点到某一个节点,路径上经过的字符连接起来,为该节点对应的字符串。
③每个节点的所有子节点包含的字符互不相同。
④从第一字符开始有连续重复的字符只占用一个节点,比如上面的to,和ten,中重复的单词t只占用了一个节点。

前缀树的作用

  1. 字符串是否前缀匹配:
    1.1 当前字符串的前缀在字典中已经存在
    1.2 当前字符串在字典中是其他字符串的前缀
  2. 字典是否包含当前字符串
  3. 词频统计
  4. 字符串排序

使用HashMap实现前缀树

import java.util.HashMap;

/**
 * java实现前缀树
 */
public class TrieTree {

    private class Node {
        public HashMap<Character , Node> childs;      //子结点
        public boolean isLeaf;     //当前结点是否是完整字符串(是否是叶结点)

        public Node () {
            this.isLeaf = false;
            this.childs = new HashMap<>();
        }

    }

    //根结点
    private  Node root;
    public TrieTree () {
        root = new Node();
    }

    /** Inserts a word into the trie. */
    public  void insert (String str) {
        insert(root , str);
    }
    //添加结点的过程既是添加结点的作用也是匹配前缀的过程
    private  void insert (Node root , String str) {
        if (str == null || str.length() == 0) return;
        char[] chars = str.toCharArray();
        Node cur = root;
        for (int i = 0 , length = chars.length ; i < length ; i++) {
            //不包含当前字符加入
            if (! cur.childs.containsKey(chars[i])) {
                cur.childs.put(chars[i] , new Node());
            }
            cur = cur.childs.get(chars[i]);
        }
        if (!cur.isLeaf)
            cur.isLeaf = true;
    }

    /** Returns if the word is in the trie. */
    public  boolean search (String str) {
        return search(root , str);
    }
    //遍历查找是否包含字符串
    private  boolean search (Node root , String str) {
        if (str == null || str.length() == 0) return false;
        char[] chars = str.toCharArray();
        Node cur = root;
        for (int i = 0 , length = chars.length ; i < length ; i++) {
            if (!cur.childs.containsKey(chars[i])) {
                return false;
            }
            cur = cur.childs.get(chars[i]);
        }
        return cur.isLeaf;
    }

    /** Returns if there is any word in the trie that starts with the given prefix. */
    public boolean startsWith(String str) {
        if (str == null || str.length() == 0) return false;
        char[] chars = str.toCharArray();
        Node cur = root;
        for (int i = 0 , length = chars.length ; i < length ; i++) {
            if (!cur.childs.containsKey(chars[i]))
                return false;
            cur = cur.childs.get(chars[i]);
        }
        return true;
    }
}

你可能感兴趣的:(数据结构与算法)