LWC 49:676. Implement Magic Dictionary

传送门:676. Implement Magic Dictionary

Problem:

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:

  • You may assume that all the inputs are consist of lowercase letters a-z.
  • For contest purpose, the test data is rather small by now. You could think about highly efficient algorithm after the contest.
  • 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.

无脑做法:
用MAP做长度的一个映射,把长度相同的放入一个LIST中,接着判断LIST中是否存在编辑距离为1的单词。

代码如下:

class MagicDictionary {

    /** Initialize your data structure here. */

    Map<Integer, List<String>> map;
    public MagicDictionary() {
        map = new HashMap<>();
    }

    /** Build a dictionary through a list of words */
    public void buildDict(String[] dict) {
        for (String word : dict) {
            map.computeIfAbsent(word.length(), k -> new ArrayList<>()).add(word);
        }
    }

    /** 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) {
        int len = word.length();
        if (!map.containsKey(len)) return false;
        for (String dict : map.get(word.length())) {
            if (valid(dict, word)) return true;
        }
        return false;
    }

    public boolean valid(String word, String dict) {
        int n = word.length();
        int j = 0;
        for (int i = 0; i < n; ++i) {
            if (word.charAt(i) == dict.charAt(i)) j++;
        }
        return j == n - 1;
    }

}