字典树 java实现

最新推荐文章于 2024-04-01 21:20:30 发布

努力上进的Mr_wang

最新推荐文章于 2024-04-01 21:20:30 发布

阅读量448

点赞数

CC 4.0 BY-SA版权

分类专栏：笔试题

本文链接：https://blog.youkuaiyun.com/w535921917/article/details/87868970

笔试题专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用Java实现的Trie树数据结构，详细展示了如何通过Trie树进行单词的存储与查询，包括节点的定义、插入操作及查询功能。Trie树是一种用于存储字符串的高效数据结构，特别适用于自动补全、拼写检查等场景。

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



import java.util.Map;
import java.util.TreeMap;

public class Trie {

    private int size;
    private TrieNode root;

    public Trie() {
        size = 0;
        root = new TrieNode();
    }

    public void put(String word) {
        TrieNode curr = root;
        for (int i = 0; i < word.length(); i++) {
            char c = word.charAt(i);
            if (!curr.next.containsKey(c)) {
                curr.next.put(c, new TrieNode());
            }
            curr = curr.next.get(c);
        }
        //如果不是已经存在的word
        if (!curr.isWord) {
            curr.isWord = true;
            size++;
        }

    }

    public boolean query(String word) {
        if (word == null || "".equals(word)) {
            return false;
        }
        TrieNode curr = root;
        for (int i = 0; i < word.length(); i++) {
            char c = word.charAt(i);
            if(!curr.next.containsKey(c)){
                return false;
            }
            curr = curr.next.get(c);
        }
        return curr.isWord;

    }

    class TrieNode {
        boolean isWord;
        Map<Character, TrieNode> next;

        public TrieNode() {
            this(false);
        }

        public TrieNode(Boolean isWord) {
            this.isWord = isWord;
            next = new TreeMap<Character, TrieNode>();
        }
    }

    public static void main(String[] args) {
        Trie trie = new Trie();
        trie.put("word");
        trie.put("world");
        trie.put("one");
        trie.put("own");
        trie.put("wow");

        System.out.println(trie.query("word"));
        System.out.println(trie.query("wod"));
        System.out.println(trie.query("d"));

    }

}