已解答
中等
相关标签
相关企业
给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2 输出: [1,2]
示例 2:
输入: nums = [1], k = 1 输出: [1]
提示:
1 <= nums.length <= 105
k
的取值范围是[1, 数组中不相同的元素的个数]
- 题目数据保证答案唯一,换句话说,数组中前
k
个高频元素的集合是唯一的
//大根堆
class Solution {
public int[] topKFrequent(int[] nums, int k) {
Map<Integer,Integer>map = new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);//从界面复制的说明: Map,返回指定键映射到的值,如果此映射不包含键的映射,则返回 defaultValue
}
//使用大根堆,对所有map里的元素排序(大根堆书写简单但是,不如小根堆的效率高)
PriorityQueue<int[]> qp = new PriorityQueue<>((pair1,pair2)->pair2[1]-pair1[1]);//lamada表达式pair2[1]-pair1[1]表示对放入的元素进行大根堆排序
for(Map.Entry<Integer,Integer>entry:map.entrySet()){
qp.add(new int[]{entry.getKey(),entry.getValue()});
}
//选取前k个元素
int[]res = new int[k];
for(int i =0;i<k;i++){
res[i]=qp.poll()[0];
}
return res;
}
}
//小根堆
class Solution {
public int[] topKFrequent(int[] nums, int k) {
Map<Integer,Integer> map = new HashMap<>();//键值对<int,int>
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
PriorityQueue<int[]>pq = new PriorityQueue<>((pair1,pair2)->pair1[1]-pair2[1]);//按小根堆排序
for(Map.Entry<Integer,Integer>entry:map.entrySet()){//遍历map
if(pq.size()<k){
pq.add(new int[]{entry.getKey(),entry.getValue()});//加入int[]数组,数组内容为<key,value>
}else{
if(entry.getValue()>pq.peek()[1]){//比较值value,进来的比pq里的大,就把堆顶弹出
pq.poll();
pq.add(new int[]{entry.getKey(),entry.getValue()});//加入新元素
}
}
}
int[] res = new int[k];
for(int i = 0;i<k;i++){
res[i] = pq.poll()[0];
}
return res;
}
}