力扣之实现一个魔法字典（python）

题目

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

实现 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

解题思路

初始化的时候我们构建一个字典，然后按找给的的列表中的字符串的长度为key来构建字典。在之后的search中我们先判断是否有相同长度的字符串在字典中，没有则返回false，有则进行下一步。根据所给字符的长度从字典中的到列表，在用所给字符与列表中的每个字符进行比较，若有大于两个字符不同，则直接返回false，若只有一个不同，则返回true

代码

class MagicDictionary(object):

    def __init__(self):
        self.dict = dict()

    def buildDict(self, dictionary):
        """
        :type dictionary: List[str]
        :rtype: None
        """
        for i in dictionary:
            i_len = len(i)
            if i_len in self.dict:
                self.dict[i_len].append(i)
            else:
                self.dict[i_len] = [i]




    def search(self, searchWord):
        """
        :type searchWord: str
        :rtype: bool
        """
        b = searchWord
        b_len = len(b)
        if b_len not in self.dict:
            return False
        a_all = self.dict[b_len]
        for a in a_all:
            if sum(1 for i in range(b_len) if a[i] != b[i]) == 1:
                return True
        return False



# Your MagicDictionary object will be instantiated and called as such:
# obj = MagicDictionary()
# obj.buildDict(dictionary)
# param_2 = obj.search(searchWord)

结果