要求
在给定单词列表 wordlist 的情况下,我们希望实现一个拼写检查器,将查询单词转换为正确的单词。
对于给定的查询单词 query,拼写检查器将会处理两类拼写错误:
-
大小写:如果查询匹配单词列表中的某个单词(不区分大小写),则返回的正确单词与单词列表中的大小写相同。
- 例如:wordlist = ["yellow"], query = "YellOw": correct = "yellow"
- 例如:wordlist = ["Yellow"], query = "yellow": correct = "Yellow"
- 例如:wordlist = ["yellow"], query = "yellow": correct = "yellow"
-
元音错误:如果在将查询单词中的元音(‘a’、‘e’、‘i’、‘o’、‘u’)分别替换为任何元音后,能与单词列表中的单词匹配(不区分大小写),则返回的正确单词与单词列表中的匹配项大小写相同。
- 例如:wordlist = ["YellOw"], query = "yollow": correct = "YellOw"
- 例如:wordlist = ["YellOw"], query = "yeellow": correct = "" (无匹配项)
- 例如:wordlist = ["YellOw"], query = "yllw": correct = "" (无匹配项) 此外,拼写检查器还按照以下优先级规则操作:
-
当查询完全匹配单词列表中的某个单词(区分大小写)时,应返回相同的单词。
-
当查询匹配到大小写问题的单词时,您应该返回单词列表中的第一个这样的匹配项。
-
当查询匹配到元音错误的单词时,您应该返回单词列表中的第一个这样的匹配项。
-
如果该查询在单词列表中没有匹配项,则应返回空字符串。 给出一些查询 queries,返回一个单词列表 answer,其中 answer[i] 是由查询 query = queries[i] 得到的正确单词。
示例:
输入:wordlist = ["KiTe","kite","hare","Hare"], queries = ["kite","Kite","KiTe","Hare","HARE","Hear","hear","keti","keet","keto"]
输出:["kite","KiTe","KiTe","Hare","hare","","","KiTe","","KiTe"]
提示:
- 1 <= wordlist.length <= 5000
- 1 <= queries.length <= 5000
- 1 <= wordlist[i].length <= 7
- 1 <= queries[i].length <= 7
- wordlist 和 queries 中的所有字符串仅由英文字母组成。
核心代码
class Solution:
def spellchecker(self, wordlist: List[str], queries: List[str]) -> List[str]:
word_set = set(wordlist)
word_dict = collections.defaultdict(list)
vowel_dict = collections.defaultdict(list)
for word in wordlist:
lower = word.lower()
word_dict[lower].append(word)
vowel_dict[re.sub("[aeiou]","#",lower)].append(word)
ans = []
for query in queries:
if query in word_set:
ans.append(query)
else:
lower = query.lower()
if lower in word_dict:
ans.append(word_dict[lower][0])
else:
vowel = re.sub("[aeiou]","#",lower)
if vowel in vowel_dict:
ans.append(vowel_dict[vowel][0])
else:
ans.append("")
return ans
解题思路:我们制作两个字典,一个字典,将所有的wordlist变成全小写,作为"全等"判断的键,值是原始的值,我们在准备下一个字典,将元音都用"#"进行替换,进行除元音外的"框架"匹配,最后循环遍历queries,先进行"全等"的匹配,然后在进行"框架"匹配,最后返回推荐单词的列表。