Leetcode 211.添加与搜索单词

添加与搜索单词

设计一个支持以下两种操作的数据结构：

void addWord(word)

bool search(word)

search(word) 可以搜索文字或正则表达式字符串，字符串只包含字母 . 或 a-z 。 . 可以表示任何一个字母。

示例:

addWord("bad")

addWord("dad")

addWord("mad")

search("pad") -> false

search("bad") -> true

search(".ad") -> true

search("b..") -> true

说明:

你可以假设所有单词都是由小写字母 a-z 组成的。

public class WordDictionary{
    public static class TrieNode{
        TrieNode[] children;
        boolean isLeaf;
        public TrieNode(){
            children=new TrieNode[26];
        }
    }

    private TrieNode root=new TrieNode();
    public void addWord(String word){
        if(word==null||word.isEmpty()) return;
        TrieNode node=root;
        for(int i=0;i<word.length();i++){
            int j=word.charAt(i)-'a';
            if(node.children[j]==null){
                node.children[j]=new TrieNode();
            }
            node=node.children[j];
        }
        node.isLeaf=true;
    }

    public boolean search(String word){
        return search(root,word,0);
    }

    private boolean search(TrieNode node,String word,int start){
        if(node==null) return false;
        if(start==word.length()) return node.isLeaf;
        for(int i=start;i<word.length();i++){
            char c=word.charAt(i);
            if(c=='.'){
                for(int j=0;j<26;j++){
                    if(search(node.children[j],word,i+1)) return true;
                }
                return false;
            }else{
                return search(node.children[c-'a'],word,i+1);
            }
        }
        return false;
    }
}