题目链接
https://leetcode-cn.com/problems/top-k-frequent-words/description/
描述
给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
注意:
假定 k 总为有效值, 1 ≤ k ≤ 集合元素数。
输入的单词均由小写字母组成。
扩展练习:
尝试以 O(n log k) 时间复杂度和 O(n) 空间复杂度解决。
示例
示例 1:
输入: ["i", "love", "leetcode", "i", "love", "coding"], k = 2
输出: ["i", "love"]
解析: "i" 和 "love" 为出现次数最多的两个单词,均为2次。
注意,按字母顺序 "i" 在 "love" 之前。
示例 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 次。
初始代码模板
class Solution {
public List<String> topKFrequent(String[] words, int k) {
}
}
代码
其实这个时间复杂度基本可以看出来是用堆,毕竟堆大小有关。
这次推荐官方题解:
https://leetcode-cn.com/problems/top-k-frequent-words/solution/qian-kge-gao-pin-dan-ci-by-leetcode/
class Solution {
public List<String> topKFrequent(String[] words, int k) {
Map<String, Integer> map = new HashMap<>();
for (int i = 0; i < words.length; i++) {
map.put(words[i], map.getOrDefault(words[i], 0) + 1);
}
PriorityQueue<String> queue = new PriorityQueue<>(
(o1, o2) -> {
return map.get(o1) == map.get(o2) ?
o2.compareTo(o1) : map.get(o1) - map.get(o2);
}
);
/*
for (String str : map.keySet()) {
if (queue.size() < k) {
queue.offer(str);
continue;
}
if (map.get(queue.peek()) < map.get(str) || (map.get(queue.peek()) == map.get(str) && queue.peek().compareTo(str) > 0)) {
queue.poll();
queue.offer(str);
}
}
*/
for (String str : map.keySet()) {
queue.offer(str);
if (queue.size() > k) {
queue.poll();
}
}
LinkedList<String> res = new LinkedList<>();
while (!queue.isEmpty()) {
res.addFirst(queue.poll());
}
return res;
}
}