LeetCode347前 K 个高频元素(哈希表加优先队列)

最新推荐文章于 2022-01-20 17:28:09 发布

我会jvav啊

最新推荐文章于 2022-01-20 17:28:09 发布

阅读量114

点赞数 1

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_43434328/article/details/114551321

版权

队列 queue 数据结构 leetcode 算法

leetcode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

题目
在这里插入图片描述
TopK问题非常常见今天学习了其中的一种解法最小堆也就是优先队列
首先我们一遍哈希存下每个数字出现的次数
然后维护一个长度为K的优先队列
小于K时直接入队等于K时判断是否大于队首元素(由于是升序队列所以大于最小的即可)

    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer,Integer> map = new HashMap<>();
        for (int n:nums) {
            if (map.containsKey(n))
                map.put(n,map.get(n)+1);
            else
                map.put(n,1);
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>((a,b) -> map.get(a)-map.get(b));
        for (int key: map.keySet()) {
            if (queue.size() < k)
                queue.add(key);
            else if (map.get(key) > map.get(queue.peek())){
                queue.poll();
                queue.add(key);
            }
        }
        int res[] = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = queue.poll();
        }
        return res;
    }