leetcode_966 元音拼写检查器

84 阅读2分钟

要求

在给定单词列表 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

image.png

解题思路:我们制作两个字典,一个字典,将所有的wordlist变成全小写,作为"全等"判断的键,值是原始的值,我们在准备下一个字典,将元音都用"#"进行替换,进行除元音外的"框架"匹配,最后循环遍历queries,先进行"全等"的匹配,然后在进行"框架"匹配,最后返回推荐单词的列表。