字典树 04 字典树中的前缀问题

最新推荐文章于 2024-12-29 17:23:05 发布

原创最新推荐文章于 2024-12-29 17:23:05 发布 · 144 阅读

0 ·

CC 4.0 BY-SA版权

数据结构 [Java] 专栏收录该内容

86 篇文章

订阅专栏

本文详细介绍了如何使用Java实现Trie(前缀树)，包括插入单词、搜索单词和查询前缀三个核心操作。Trie是一种用于高效存储和检索字符串的树形数据结构，特别适合用于自动补全和拼写检查等场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

查询是否在Trie中有单词以prefix为前缀

prefix在Trie中都找到的情况下，不用再判断prefix.charAt(prefix.length - 1)所对应的节点是否是单词，直接返回true即可；

// 查询是否在Trie中有单词以prefix为前缀
public boolean isPrefix(String prefix){

    Node cur = root;
    for(int i = 0 ; i < prefix.length() ; i ++){
        char c = prefix.charAt(i);
        if(cur.next.get(c) == null)
            return false;
        cur = cur.next.get(c);
    }

    return true;
}

A Problem In Leetcode

208. 实现 Trie (前缀树)

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;

    public Trie(){
        root = new Node();
    }

    // 向Trie中添加一个新的单词word
    public void insert(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);
        }
        cur.isWord = true;
    }

    // 查询单词word是否在Trie中
    public boolean search(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);
        }
        return cur.isWord;
    }

    // 查询是否在Trie中有单词以prefix为前缀
    public boolean startsWith(String isPrefix){

        Node cur = root;
        for(int i = 0 ; i < isPrefix.length() ; i ++){
            char c = isPrefix.charAt(i);
            if(cur.next.get(c) == null)
                return false;
            cur = cur.next.get(c);
        }

        return true;
    }
}

/**
 * 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);
 */