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

leetcode-347 前k个高频元素

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

leetcode-347 前k个高频元素

优先队列 题目描述:

给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

测试样例

示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]

解题思路:

先将每一个元素的频率用HashMap存储起来,再构建一个小顶堆用来根据元素频率排序,每一次添加都淘汰堆顶最小的元素,最后剩下的就是出现频率最高的k个元素。

难点:

PriorityQueue的创建,PriorityQueue里要存储Entry对象。
PriorityQueue> deque = new PriorityQueue<>((o1,o2) -> o1.getValue() - o2.getValue());

代码实现:
class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map map = new HashMap<>();
        int len = nums.length;
        for(int i = 0;i < len;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0) + 1);
        }
        Set> entris = map.entrySet();
        PriorityQueue> deque = new PriorityQueue<>((o1,o2) -> o1.getValue() - o2.getValue());
        for(Map.Entry entry : entris){
            deque.offer(entry);
            if(deque.size() > k){
                deque.poll();
            }
        }
        int[] res = new int[k];
        for(int i = k - 1;i >= 0;i--){
            res[i] = deque.poll().getKey();
        }
        return res;
    }
}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/878570.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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