LeetCode819. 最常见的单词（python）

题目

给定一个段落 (paragraph) 和一个禁用单词列表 (banned)。返回出现次数最多，同时不在禁用列表中的单词。
题目保证至少有一个词不在禁用列表中，而且答案唯一。
禁用列表中的单词用小写字母表示，不含标点符号。段落中的单词不区分大小写。答案都是小写字母。
示例：
输入:
paragraph = “Bob hit a ball, the hit BALL flew far after it was hit.”
banned = [“hit”]
输出: “ball”
解释:
“hit” 出现了3次，但它是一个禁用的单词。
“ball” 出现了2次 (同时没有其他单词出现2次)，所以它是段落里出现次数最多的，且不在禁用列表中的单词。
注意，所有这些单词在段落里不区分大小写，标点符号需要忽略（即使是紧挨着单词也忽略， 比如 “ball,”），
"hit"不是最终的答案，虽然它出现次数更多，但它在禁用单词列表中。

思路

先处理符号，再计算单词出现次数，最后取出不在禁用单词里且出现次数最大的单词

class Solution:
    def mostCommonWord(self, paragraph: str, banned: List[str]) -> str:     
        # 处理特殊符号
        for i in "!?',;.":
            if i in paragraph:
                # 把所有的符号换成空格
                paragraph = paragraph.replace(i, ' ').replace('  ', ' ')        
        # 将paragraph转为小写字母，去掉首位的空格，再转换成单词列表
        res = list(paragraph.lower().strip().split(' '))
        dicts = {}
        # 遍历所有单词遍，转换成单词为key,出现次数为value的字典
        for re in res:
            if re in dicts.keys():
                dicts[re] += 1
            else:
                dicts[re] = 1
        # 去除字典中的的禁用单词
        for key in list(dicts.keys()):
            if key in banned:
                del dicts[key]
        # 返回字典中value最大值的key
        return list(dicts.keys())[list(dicts.values()).index(max(dicts.values()))]

题解

优化一下，计算每个单词出现次数的同时，不计入禁用单词出现的次数
最后直接去出现次数最大的单词就行

class Solution:
    def mostCommonWord(self, paragraph: str, banned: List[str]) -> str:
        for i in "!?',;.":
            if i in paragraph:
                paragraph = paragraph.replace(i, ' ').replace('  ', ' ')
        res = list(paragraph.lower().strip().split(' '))
        dicts = {}
        # 将转换单词出现次数字典和去除字典中的的禁用单词同步进行
        for re in res:
            if re not in banned:
                if re in dicts.keys():
                    dicts[re] += 1
                else:
                    dicts[re] = 1
        # 用value列表求出最大值的索引，用该索引取去key列表的值
        return list(dicts.keys())[list(dicts.values()).index(max(dicts.values()))]

过啦~