跳过正文
  1. leetcode 题解/

676_实现一个魔法字典

·236 字·2 分钟
目录

题目
#

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

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

解法
#

将 words 存储在字典树中。

在搜索过程中,我们尝试改变当前位置的字符(之前没改过的情况下)之后继续进行搜索。

代码
#

class MagicDictionary:

    def __init__(self):
        self.root = {'end': False}

    def insert(self, word):
        node = self.root
        for c in word:
            if c not in node:
                node[c] = {'end': False}
            node = node[c]
        node['end'] = True

    def buildDict(self, dictionary: List[str]) -> None:
         for word in dictionary:
            self.insert(word)

    def doSearch(self, word, i, node, changed):
        if i == len(word):
            return changed and node and node['end']
        c = word[i]
        if c not in node and changed:
            return False
        return (c in node and self.doSearch(word, i+1, node[c], changed) ) or (not changed and any([ x!= 'end' and x != c and self.doSearch(word, i+1, node[x], True) for x in node.keys() ]) )

    def search(self, searchWord: str) -> bool:
        return self.doSearch(searchWord, 0, self.root, False)