【LeetCode】第692题——前K个高频单词（难度：中等）

题目描述

给一非空的单词列表，返回前 k 个出现次数最多的单词。

返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率，按字母顺序排序。

1. 示例 1：
   输入: [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
   输出: [“i”, “love”]
   解析: “i” 和 “love” 为出现次数最多的两个单词，均为2次。注意，按字母顺序 “i” 在 “love” 之前。

2. 示例 2：
   输入: [“the”, “day”, “is”, “sunny”, “the”, “the”, “the”, “sunny”, “is”, “is”], k = 4
   输出: [“the”, “is”, “sunny”, “day”]
   解析: “the”, “is”, “sunny” 和 “day” 是出现次数最多的四个单词，出现次数依次为 4, 3, 2 和 1 次。

注意：
假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
输入的单词均由小写字母组成。

扩展练习：
尝试以 O(n log k) 时间复杂度和 O(n) 空间复杂度解决。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/top-k-frequent-words
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

先用一个HashMap记录每个单词出现的频次，然后用比较器进行比较排序即可

代码详解

class Solution {
    public List<String> topKFrequent(String[] words, int k) {
        HashMap<String, Integer> hm = new HashMap<>(); // 记录单词出现频次
        for(String s : words) {
            hm.put(s, hm.getOrDefault(s, 0) + 1);
        }
        List<String> list = new ArrayList<>(); // 需要进行排序的String列表
        for(String s : hm.keySet()) {
            list.add(s);
        }
        Collections.sort(list, (a, b) -> (hm.get(a) == hm.get(b) ? a.compareTo(b) : hm.get(b) - hm.get(a)));
        // Lambda表达式，注意最好是数量降序排序，数量一样的话，字母升序排序
        // 意思是如果字符串a和字符串b的出现频次一样的话，用compareTo方法，小的放前面，用a.compareTo(b)
        // 如果出现频次不一样，返回hm.get(b) - hm.get(a)，出现频次降序排序
        return list.subList(0, k);
    }
}

注意点

• 此题的难点仅在于排序方法如何编写