leetcode 347 前K个高频元素

最新推荐文章于 2022-04-22 10:22:35 发布

斜月&三星

最新推荐文章于 2022-04-22 10:22:35 发布

阅读量280

点赞数

CC 4.0 BY-SA版权

分类专栏： leetecode

本文链接：https://blog.youkuaiyun.com/qq_43322057/article/details/84842463

leetecode 专栏收录该内容

145 篇文章

订阅专栏

本文介绍了一种高效查找数组中出现频率最高的k个元素的方法。通过使用HashMap记录每个元素的出现次数，然后利用PriorityQueue（最大堆）进行排序，从而快速找到目标元素。此算法适用于大数据集的统计分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

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

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

思路：首先应该考虑用HashMap来做，建立数字和其出现次数的映射，然后再按照出现次数进行排序。我们可以用堆排序来做，使用一个最大堆来按照映射次数从大到小排列，在java中使用PriorityQueue来实现。

public List<Integer> topKFrequent(int[] nums, int k) {
    //第一步:存放值 和值出现的次数
    Map<Integer,Integer> map=new HashMap<>();
    for (int num : nums) {
        if(map.get(num) == null){
            map.put(num, 1);
        }else{
            map.put(num, map.get(num) + 1);
        }
    }
    //第二步：生成优先队列
    PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
        @Override
        public int compare(Integer a, Integer b) {
            return map.get(b) - map.get(a);
	    }
	});
    for(Integer key : map.keySet()){
        queue.add(key);
    }
    List<Integer> list = new ArrayList<>();
    for(int i = 0; i < k; i++){
        list.add(queue.remove());
    }
    return list;
}