2020-11-16 676. 实现一个魔法字典

设计一个使用单词列表进行初始化的数据结构，单词列表中的单词 互不相同 。 如果给出一个单词，请判定能否只将这个单词中一个字母换成另一个字母，使得所形成的新单词存在于你构建的字典中。

实现 MagicDictionary 类：

    MagicDictionary() 初始化对象
    void buildDict(String[] dictionary) 使用字符串数组 dictionary 设定该数据结构，dictionary 中的字符串互不相同
    bool search(String searchWord) 给定一个字符串 searchWord ，判定能否只将字符串中 一个 字母换成另一个字母，使得所形成的新字符串能够与字典中的任一字符串匹配。如果可以，返回 true ；否则，返回 false 。

 

示例：

输入
["MagicDictionary", "buildDict", "search", "search", "search", "search"]
[[], [["hello", "leetcode"]], ["hello"], ["hhllo"], ["hell"], ["leetcoded"]]
输出
[null, null, false, true, false, false]

解释
MagicDictionary magicDictionary = new MagicDictionary();
magicDictionary.buildDict(["hello", "leetcode"]);
magicDictionary.search("hello"); // 返回 False
magicDictionary.search("hhllo"); // 将第二个 'h' 替换为 'e' 可以匹配 "hello" ，所以返回 True
magicDictionary.search("hell"); // 返回 False
magicDictionary.search("leetcoded"); // 返回 False

 

提示：

    1 <= dictionary.length <= 100
    1 <= dictionary[i].length <= 100
    dictionary[i] 仅由小写英文字母组成
    dictionary 中的所有字符串 互不相同
    1 <= searchWord.length <= 100
    searchWord 仅由小写英文字母组成
    buildDict 仅在 search 之前调用一次
    最多调用 100 次 search

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/implement-magic-dictionary
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

前缀树+DFS

class MagicDictionary {
public:
    bool isEnd;
    MagicDictionary* next[26];
public:
    /** Initialize your data structure here. */
    MagicDictionary() {
        isEnd = false;
        memset(next, 0, sizeof(next));
    }
    
    //创建前缀树
    void buildDict(vector<string> dictionary) {
        for(string str : dictionary) {
            MagicDictionary* node = this;
            for(char c : str) {
                if(node->next[c - 'a'] == NULL) {
                    node->next[c - 'a'] = new MagicDictionary();
                }
                node = node->next[c - 'a'];
            }
            node->isEnd = true;
        }
    }
    
    //查找, 替换一个字母，使得在字典中，如果完全相同的话，直接返回false
    bool search(string searchWord) {
        if(searchWord.empty()) {
            return false;
        }
        MagicDictionary* node = this;
        bool used = false;
        return dfs(searchWord, node, 0, used);
    }

    bool dfs(string searchWord, MagicDictionary* node, int start, bool used) {
        if(start == searchWord.size()) {
            return node->isEnd && used;
        } 
        char temp = searchWord[start];
        if(node->next[temp - 'a'] == NULL) {
            if(used) {
                return false;
            } else {
                for(int i = 0; i < 26; i ++) {
                    if(node->next[i] != NULL && dfs(searchWord, node->next[i], start + 1, true)) {
                        return true;
                    } 
                }
                return false;
            }
        } else {
            if(dfs(searchWord, node->next[temp - 'a'], start + 1, used)) {
                return true;
            } 
            if(!used) {
                for(int i = 0; i < 26; i ++) {
                    if(i != temp - 'a' && node->next[i] && dfs(searchWord, node->next[i], start + 1, true)) {
                        return true;
                    }
                }
                return false;
            }
        }
        return false;
    }
};

/**
 * Your MagicDictionary object will be instantiated and called as such:
 * MagicDictionary* obj = new MagicDictionary();
 * obj->buildDict(dictionary);
 * bool param_2 = obj->search(searchWord);
 */