前k个高频元素_优先队列_java

前k个高频元素

leetcode链接

问题描述

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

提示:

1 <= nums.length <= 10^5

k 的取值范围是 [1, 数组中不相同的元素的个数]

题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的

示例

示例 1:

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

输出: [1,2]

示例 2:

输入: nums = [1], k = 1

输出: [1]

解题思路

  1. 用map记录数组中元素出现的频率

    1. 将map中元素都存入PriorityQueue,优先队列会维护队列内元素以大根堆(或小根堆)的形式排序
    2. 从优先队列中取k个元素的key,即为数组的前K个高频元素

代码实现

java 复制代码
class Solution {
   public int[] topKFrequent(int[] nums, int k) {

       Map<Integer, Integer> map = new HashMap<>();

       //用map记录数组中元素出现的频率
       for (Integer num : nums) {
           map.put(num, map.getOrDefault(num,0)+1);
       }

       //创建一个以大根堆形式存储队内元素的优先队列
       PriorityQueue<int[]> pq = new PriorityQueue<>((pair1, pair2) -> pair2[1] - pair1[1]);

       //map中元素以数组形式存入PriorityQueue
       for (Map.Entry entry : map.entrySet()) {
           //注意:此处需要进行强制类型转换,因为entry.getKey()与entry.getValue()的返回类型是Object!!!
           pq.add(new int[]{(Integer)entry.getKey(),(Integer)entry.getValue()});
       }

       //从优先级队列中取k个元素的key,并加入结果集中
       int[] res = new int[k];
       for (int i = 0;  i < k; i++) {
           res[i] = pq.poll()[0];
       }

       return res;


   }
}

复杂度

时间复杂度:

PriorityQueue类的add方法的时间复杂度为O(logn), 调用了n次所以时间复杂度为O(nlogn)

空间复杂度:

度:

PriorityQueue类的add方法的时间复杂度为O(logn), 调用了n次所以时间复杂度为O(nlogn)

空间复杂度:

O(n)

相关推荐
SimonLiu0099 分钟前
清理HiNas(海纳斯) Docker日志并限制日志大小
java·docker·容器
带刺的坐椅10 分钟前
开发 MCP Proxy(代理)也可以用 Solon AI MCP 哟!
java·ai·llm·solon·mcp·mcp-server·mcp-client
yuren_xia25 分钟前
Spring XML 配置
xml·java·spring
每次的天空31 分钟前
kotlin与MVVM结合使用总结(三)
开发语言·microsoft·kotlin
mit6.82439 分钟前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify42 分钟前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法
小鸡脚来咯42 分钟前
SpringBoot 常用注解大全
java
ephemerals__1 小时前
【c++11】c++11新特性(下)(可变参数模板、default和delete、容器新设定、包装器)
开发语言·c++
shix .1 小时前
2025年PTA天梯赛正式赛 | 算法竞赛,题目详解
数据结构·算法
先生沉默先1 小时前
c#接口_抽象类_多态学习
开发语言·学习·c#