211. Add and Search Word - Data structure design

一、题目

　　1、审题

　　

　　2、分析

　　　　与208不同的是， search() 方法可以传入的单词可以包含 '.' ，代表任意一个字符。

 

二、解答

　　1、思路：

　　　　采用 DFS 方式，当查找单词中字符为 '.' 时，从当前 Trie 节点的所有非空子节点开始查找一次，若有一个返回成功，则为 true；若都失败，则 false；

class WordDictionary {

    private TrieNode root;
    /** Initialize your data structure here. */
    public WordDictionary() {
        root = new TrieNode();
    }
    
    /** Adds a word into the data structure. */
    public void addWord(String word) {
        TrieNode node = root;
        for (int i = 0; i < word.length(); i++) {
            char c = word.charAt(i);
            if(node.children[c - 'a'] == null)
                node.children[c - 'a'] = new TrieNode();
            node = node.children[c - 'a'];
        }
        node.isWord = true;
    }
    
    /** Returns if the word is in the data structure. A word could contain the dot character '.' to represent any one letter. */
    public boolean search(String word) {
        return search(word, root);
    }

    private boolean search(String word, TrieNode node) {
        for (int i = 0; i < word.length(); i++) {
            char c = word.charAt(i);
            // 处理 '.'
            if(c == '.') {
                for(TrieNode child: node.children) {
                    if(child == null)
                        continue;
                    if(search(word.substring(i+1), child))
                        return true;
                }
                return false;
            }
            if(node.children[c - 'a'] == null)
                return false;
            node = node.children[c - 'a'];
        }
        return node.isWord;
    }
}