设计一个使用单词列表进行初始化的数据结构,单词列表中的单词 互不相同 。 如果给出一个单词,请判定能否只将这个单词中一个字母换成另一个字母,使得所形成的新单词存在于已构建的神奇字典中。 实现 MagicDictionary 类: MagicDictionary() 初始化对象 void buildDict(String[] dictionary) 使用字符串数组 dictionary 设定该数据结构,dictionary 中的字符串互不相同 bool search(String searchWord) 给定一个字符串 searchWord ,判定能否只将字符串中 一个 字母换成另一个字母,使得所形成的新字符串能够与字典中的任一字符串匹配。如果可以,返回 true ;否则,返回 false 。 示例: 输入 inputs = ["MagicDictionary", "buildDict", "search", "search", "search", "search"] inputs = [[], [["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
逻辑思路: 按照题目的意思,首先必须是长度一致的字符串才有可能,长度大于或小于搜索单词的都可以排除;对比长度相等的两个字符串,只有一个不相同的字典单词就是答案。把长度相等的字典单词划为一个词组,搜索某个单词时,就在这个搜索单词的对应长度的词组里面查找。
代码:
"""
@author: MR.N
@created: 2021/9/17 11:45
@blog: https://blog.csdn.net/qq_21264377
@description: 神奇的字典.
"""
__author__ = 'MR.N'
class MagicDictionary:
def __init__(self):
"""
Initialize your data structure here.
"""
self.dic = {}
def buildDict(self, dictionary) -> None:
for word in dictionary:
# 以输入单词的长度作为键,划分字典词组
key = str(len(word))
# 判断是否存在输入单词长度的词组
# 否则,新建一个且加入该单词
if key not in self.dic:
self.dic[key] = [word]
else:
if word not in self.dic.get(key):
self.dic.get(key).append(word)
def search(self, searchWord: str) -> bool:
key = str(len(searchWord))
local_dic = self.dic.get(key, None)
# 相等长度的词组不存在
if local_dic is None:
return False
for word in local_dic:
diff = 0
for i in range(len(word)):
# 不止一个不同字符
if diff > 1:
break
if word[i] != searchWord[i]:
diff += 1
# 只有一个不同字符
if diff == 1:
return True
return False
注释:
1)题目来源于Leetcode力扣中国 -- leetcode-cn.com