Leetcode 208.实现前缀树

实现前缀树

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert("apple");

trie.search("apple"); // 返回 true

trie.search("app"); // 返回 false

trie.startsWith("app"); // 返回 true

trie.insert("app");

trie.search("app"); // 返回 true

说明:

• 你可以假设所有的输入都是由小写字母 a-z 构成的。
• 保证所有输入均为非空字符串。

dclass TrieNode{
        char c;
        boolean leaf;
        HashMap<Character,TrieNode> children=new HashMap<Character,TrieNode>();
public TrieNode(char c){
        this.c=c;
        }
public TrieNode(){};
        }

public class Trie{
    private TrieNode root;
    public Trie(){
        root=new TrieNode();
    }

    public void insert(String word){
        Map<Character,TrieNode> children=root.children;
        for(int i=0;i<word.length();i++){
            char c=word.charAt(i);
            TrieNode t;
            if(children.containsKey(c)){
                t=children.get(c);
            }else{
                t=new TrieNode(c);
                children.put(c,t);
            }
            children=t.children;
            if(i==word.length()-1)
                t.leaf=true;
        }
    }

    public boolean search(String word){
        TrieNode t=searchNode(word);
        return t!=null && t.leaf;
    }

    private TrieNode searchNode(String word){
        Map<Character,TrieNode> children=root.children;
        TrieNode t=null;
        for(int i=0;i<word.length();i++){
            char c=word.charAt(i);
            if(!children.containsKey(c)) return null;
            t=children.get(c);
            children=t.children;
        }
        return t;
    }

    public boolean startsWith(String prefix){
        return searchNode(prefix)!=null;
    }
}