LeetCode819. 最常见的单词

题目

给定一个段落 (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"不是最终的答案，虽然它出现次数更多，但它在禁用单词列表中。

说明:

• 1 <= 段落长度 <= 1000.
• 1 <= 禁用单词个数 <= 100.
• 1 <= 禁用单词长度 <= 10.
• 答案是唯一的, 且都是小写字母 (即使在 paragraph 里是大写的，即使是一些特定的名词，答案都是小写的。)
• paragraph 只包含字母、空格和下列标点符号!?',;.
• paragraph 里单词之间都由空格隔开。
• 不存在没有连字符或者带有连字符的单词。
• 单词里只包含字母，不会出现省略号或者其他标点符号。

分析

 先处理字符串，将大写转为小写，并且去掉所有的符号；然后用map统计下词频，放入PriorityQueue中，降序排序，输出除了banned中以外频率最大的单词。

代码

class Solution {
    public String mostCommonWord(String paragraph, String[] banned) {
        paragraph = paragraph.toLowerCase().replaceAll("[^a-z]"," ");
        String[] paragraphs = paragraph.split(" ");
        
        Map<String,Integer> map = new HashMap<>();
        Set<String> bannedSet = new HashSet<>();

        for (int i = 0; i < paragraphs.length; i++)
            if (!paragraphs[i].equals("") && map.containsKey(paragraphs[i])) map.put(paragraphs[i],map.get(paragraphs[i])+1);
            else if (!paragraphs[i].equals("")) map.put(paragraphs[i],1);

        for (int i = 0; i < banned.length; i++)
            bannedSet.add(banned[i]);

        PriorityQueue<Map.Entry<String,Integer>> pq = new PriorityQueue<>(new Comparator<Map.Entry<String, Integer>>() {
            @Override
            public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {
                return o2.getValue()-o1.getValue();
            }
        });

        for(Map.Entry<String,Integer> entry : map.entrySet())
            pq.add(entry);


        int len = pq.size();
        for (int i = 0; i < len; i++) {
            Map.Entry<String,Integer> temp = pq.poll();
            if (!bannedSet.contains((temp.getKey()))) return temp.getKey();
        }
        
        
        return "";
    }
}