栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

LeetCode

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

LeetCode

思路(优先队列)
  1. 首先将值和其出现的次数存储到一个哈希表中形成映射关系;
  2. 然后我们创建一个优先队列,存储的元素就是哈希表中的键值对,我们根据值(出现次数),由小到大排列,(实现原理就是小顶堆);
  3. 遍历哈希表的元素,将元素放入优先队列中,放入的过程中队列会自动根据出现次数由小到大将元素存入;我们在存入过程中就可以根据题目中的k控制放入元素的数量,题目要求前K个高频元素,所以我们就可以保持优先队列中有K个元素,当队列中元素超过K时,就讲队头出现次数较小的元素弹出,依次加入和弹出
  4. 最后遍历队列,将哈希表的键存到最终的结果集中即可
实现代码(java)
class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        linkedHashMap map = new linkedHashMap<>();
        // 将值和出现的数量存储到 哈希表 中
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }

        // 得到小顶堆的优先队列,队头存储的是出现次数较小的元素
        PriorityQueue> queue = new PriorityQueue<>((o1, o2) -> o1.getValue() - o2.getValue());

        // 遍历哈希表中的各个元素
        final Set> entries = map.entrySet();
        for (Map.Entry entry : entries) {
            queue.offer(entry);
            // 如果队列长度大于k 则将队头较小的元素弹出,这样队列中存储的就是最后k个较大的元素
            if (queue.size() > k) {
                queue.poll();
            }
        }

        // 遍历队列,得到哈希表中的键,即为前K个高频元素
        int[] res = new int[k];
        for (int i = k - 1; i >= 0; i--) {
            res[i] = queue.poll().getKey();
        }

        return res;
    }
}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/349515.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号