Algorithms—208.Implement Trie (Prefix Tree)

最新推荐文章于 2020-08-09 20:43:42 发布

原创最新推荐文章于 2020-08-09 20:43:42 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

226 篇文章

订阅专栏

本文介绍了一种使用Java实现的Trie树数据结构。该结构支持字符串的插入、搜索及前缀匹配等功能，适用于自动补全、拼写检查等应用场景。

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

思路：构造一个最多有26叉的树，为了方便，用map装子树，增加一个布尔值标志，布尔值为真，表示至此节点，有一个单词，布尔值为假，表示此节点不是某个单词的最后一个字母。

class TrieNode {
    // Initialize your data structure here.
    Map<Character,TrieNode> sons=new HashMap<Character,TrieNode>();
    char val;
    boolean flag=false;
    public TrieNode(char x) {
        val=x;
    }
}

public class Trie {
    private TrieNode root;

    public Trie() {
        root = new TrieNode(' ');
    }

    // Inserts a word into the trie.
    public void insert(String word) {
        if(!search(word)){
        	TrieNode tt=root;
            char[] ac=word.toCharArray();
            for(int i=0;i<ac.length;i++){
                Map<Character,TrieNode> sons=root.sons;
                if(sons.get(ac[i])==null){
                    TrieNode node=new TrieNode(ac[i]);
                    if(i==ac.length-1){
                        node.flag=true;
                    }
                    sons.put(ac[i],node);
                    root=node;
                }else{
                    root=sons.get(ac[i]);
                    if(i==ac.length-1){
                        root.flag=true;
                    }
                }
            }
            root=tt;
        }
    }

    // Returns if the word is in the trie.
    public boolean search(String word) {
    	TrieNode tt=root;
        char[] ac=word.toCharArray();
        for(int i=0;i<ac.length;i++){
            Map<Character,TrieNode> sons=root.sons;
            if(sons.get(ac[i])==null){
                root=tt;
                return false;
            }
            root=sons.get(ac[i]);
        }
        boolean ans=root.flag;
        root=tt;
        return ans;
    }

    // Returns if there is any word in the trie
    // that starts with the given prefix.
    public boolean startsWith(String prefix) {
    	TrieNode tt=root;
        char[] ac=prefix.toCharArray();
        for(int i=0;i<ac.length;i++){
            Map<Character,TrieNode> sons=root.sons;
            if(sons.get(ac[i])==null){
                root=tt;
                return false;
            }
            root=sons.get(ac[i]);
        }
        root=tt;
        return true;
    }
}

// Your Trie object will be instantiated and called as such:
// Trie trie = new Trie();
// trie.insert("somestring");
// trie.search("key");