【LeetCode】347. 前 K 个高频元素

最新推荐文章于 2024-10-06 18:52:52 发布

---SUN---

最新推荐文章于 2024-10-06 18:52:52 发布

阅读量121

点赞数 1

分类专栏： LeetCode 文章标签： hashmap queue 排序

本文链接：https://blog.csdn.net/SUN__CGJ/article/details/111819371

版权

LeetCode 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

347. 前 K 个高频元素

先用hashMap保存频率
再用根据频率用堆排序（最大频率堆）
获得前面的k个元素即可

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int num : nums) {
            hashMap.put(num,hashMap.getOrDefault(num, 0)+1);
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return hashMap.get(o1) > hashMap.get(o2) ? -1 : 1;
            }
        });
        int[] ans=new int[k];
        queue.addAll(hashMap.keySet());
        for (int i = 0; i < k; i++) {
            ans[i]=queue.poll();
        }
        return ans;
    }
}