题目描述
给你一个整数数组 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 个高频元素的集合是唯一的
做题记录
- 今日七刷
- 这个题要求我们求出现频率最高的前k的元素
- 核心考察:优先队列
- 注意:
- 这个题目的重点就是自定义比较器的规则
- 通过一个数组存储两个值,一个是元素的值,一个是元素的个数
- 比较的规则就是按照元素的个数从小到大排列
- 我们可以先通过Map来存储每隔元素及对应的个数
- 再往队列里面放元素的时候需要先判断一下队列元素的个数和k的关系,只有小于k才往里面放,否则要跟优先队列首元素比较个数的大小关系
代码实现
class Solution {
public int[] topKFrequent(int[] nums, int k) {
if(k>nums.length)return new int[0];
Queue<int[]> queue=new PriorityQueue<>(new Comparator<int[]>(){
public int compare(int[] a,int[] b){
return a[1]-b[1];
}
});
Map<Integer,Integer> map=new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
for(Map.Entry<Integer,Integer> entry:map.entrySet()){
int count=entry.getValue();
int num=entry.getKey();
if(queue.size()<k){
queue.offer(new int[]{num,count});
}else{
int[] n=queue.peek();
if(n[1]<count){
queue.poll();
queue.offer(new int[]{num,count});
}
}
}
int[] res=new int[k];
for(int i=0;i<k;i++){
res[i]=queue.poll()[0];
}
return res;
}
}
题目链接
题目描述
给你一个整数数组 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 个高频元素的集合是唯一的
做题记录
- 今日七刷
- 这个题要求我们求出现频率最高的前k的元素
- 核心考察:优先队列
- 注意:
- 这个题目的重点就是自定义比较器的规则
- 通过一个数组存储两个值,一个是元素的值,一个是元素的个数
- 比较的规则就是按照元素的个数从小到大排列
- 我们可以先通过Map来存储每隔元素及对应的个数
- 再往队列里面放元素的时候需要先判断一下队列元素的个数和k的关系,只有小于k才往里面放,否则要跟优先队列首元素比较个数的大小关系
代码实现
class Solution {
public int[] topKFrequent(int[] nums, int k) {
if(k>nums.length)return new int[0];
Queue<int[]> queue=new PriorityQueue<>(new Comparator<int[]>(){
public int compare(int[] a,int[] b){
return a[1]-b[1];
}
});
Map<Integer,Integer> map=new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
for(Map.Entry<Integer,Integer> entry:map.entrySet()){
int count=entry.getValue();
int num=entry.getKey();
if(queue.size()<k){
queue.offer(new int[]{num,count});
}else{
int[] n=queue.peek();
if(n[1]<count){
queue.poll();
queue.offer(new int[]{num,count});
}
}
}
int[] res=new int[k];
for(int i=0;i<k;i++){
res[i]=queue.poll()[0];
}
return res;
}
}