文章目录
- 692. 前K个高频单词
- 215. 数组中的第K个最大元素
- 2336、无限集中的最小数字
这篇文章以后记录自己刷到的题目中与堆有关的。
692. 前K个高频单词
这个题目整体不难,是前k个高频元素的改进版,只需要在创建小顶堆时执行排序规则即可。如果出现次数不同,按照出现次数由小到大排序,如果出现次数相同。**使用compareTo将字符串按照升序排序。**完成需用Collections.reverse()函数对结果反转。
代码
class Solution {
public List<String> topKFrequent(String[] words, int k) {
// build a map
HashMap<String, Integer> map = new HashMap<>();
List<String> ans = new ArrayList<>();
for(String word : words){
map.put(word, map.getOrDefault(word, 0) + 1);
}
// 小顶堆
PriorityQueue<String[]> pq = new PriorityQueue<>((a, b)-> {
if(Integer.parseInt(a[1]) != Integer.parseInt(b[1])){
return Integer.parseInt(a[1]) - Integer.parseInt(b[1]);
}
return b[0].compareTo(a[0]);
});
for(Map.Entry<String, Integer> entry : map.entrySet()){
String word = entry.getKey();
String cnts = "" + entry.getValue();
pq.add(new String[]{word, cnts});
if(pq.size() > k){
pq.poll();
}
}
while(!pq.isEmpty()){
ans.add(pq.poll()[0]);
}
// 对获取到数据进行反转
Collections.reverse(ans);
return ans;
}
}
215. 数组中的第K个最大元素
参考博客链接:https://blog.csdn.net/weixin_45863010/article/details/139702402
2336、无限集中的最小数字
参考博客链接:https://blog.csdn.net/weixin_45863010/article/details/139702402