LeetCode 676. Implement Magic Dictionary

原题链接在这里：https://leetcode.com/problems/implement-magic-dictionary/

题目：

Implement a magic directory with buildDict, and search methods.

For the method buildDict, you'll be given a list of non-repetitive words to build a dictionary.

For the method search, you'll be given a word, and judge whether if you modify exactly one character into another character in this word, the modified word is in the dictionary you just built.

Example 1:

Input: buildDict(["hello", "leetcode"]), Output: Null
Input: search("hello"), Output: False
Input: search("hhllo"), Output: True
Input: search("hell"), Output: False
Input: search("leetcoded"), Output: False

Note:

1. You may assume that all the inputs are consist of lowercase letters a-z.
2. For contest purpose, the test data is rather small by now. You could think about highly efficient algorithm after the contest.
3. Please remember to RESET your class variables declared in class MagicDictionary, as static/class variables are persisted across multiple test cases. Please see here for more details.

题解：

For each of the word in dict. Replace a char from word with "*" and use it as key, add that char to key's value set.

e.g. "hello". -> "*ello" : h.

When searching, repleace each char in the word with "*" and check if this key could be found in the dictionary.

e.g. "mheeo" -> "*ello" : m.

If it could found, check the char. If the char is not in value set or value set has it but value set also has other char, then return true.

Note: when hm contains string, and set size > 1, that means there is another string fulfilled, return true.

Time Complexity: buildDict, O(n*m). search, O(m). n = dict.length. m is average length of the word in the dictionary.

Space: O(n*m).

AC Java:

class MagicDictionary {
    HashMap<String, HashSet<Character>> hm;
    
    /** Initialize your data structure here. */
    public MagicDictionary() {
        hm = new HashMap<>();
    }
    
    /** Build a dictionary through a list of words */
    public void buildDict(String[] dict) {
        for(String word : dict){
            for(int i = 0; i<word.length(); i++){
                String sub = word.substring(0, i) + "*" + word.substring(i+1);
                if(!hm.containsKey(sub)){
                    hm.put(sub, new HashSet<Character>());
                }
                
                hm.get(sub).add(word.charAt(i));
            }
        }
    }
    
    /** Returns if there is any word in the trie that equals to the given word after modifying exactly one character */
    public boolean search(String word) {
        for(int i = 0; i<word.length(); i++){
            String sub = word.substring(0, i) + "*" + word.substring(i+1);
            if(hm.containsKey(sub)){
                HashSet<Character> hs = hm.get(sub);
                if(!hs.contains(word.charAt(i)) || hs.size()>1){
                    return true;
                }
            }
        }
        
        return false;
    }
}

/**
 * Your MagicDictionary object will be instantiated and called as such:
 * MagicDictionary obj = new MagicDictionary();
 * obj.buildDict(dict);
 * boolean param_2 = obj.search(word);
 */