给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例二:
输入: nums = [1], k = 1
输出: [1]
提示:
- 1 <= nums.length <= 105
- k 的取值范围是 [1, 数组中不相同的元素的个数]
- 题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的
进阶:你所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n 是数组大小。
思路:
选出前k的高频元素,使用小顶堆。
代码实现:
import javax.swing.undo.UndoableEdit;
import java.sql.SQLOutput;
import java.util.*;
public class Helloworld {
public int[] topKFrequent(int[] nums, int k){
Map<Integer,Integer> map = new HashMap<Integer, Integer>();
for (int num:nums) {
map.put(num,map.getOrDefault(num,0) + 1);
}
PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return map.get(o1) - map.get(o2); //升序,小顶堆
//return map.get(o1) - map.get(o2); //降序,大顶堆
}
});
for (Integer num:map.keySet()) {
System.out.println(num+" "+map.get(num));
}
for (Integer key:map.keySet()) {
if(pq.size() < k)
pq.add(key);
else if(map.get(key) > map.get(pq.peek())){
pq.poll();
pq.add(key);
}
}
int[] res = new int[k];
for (int i = 0; i < k; i++) {
res[i] = pq.poll();
}
return res;
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
String string = scanner.nextLine();
int k = scanner.nextInt();
String[] str = string.split(" ");
int[] nums = new int[str.length];
for (int i = 0; i < str.length; i++) {
nums[i] = Integer.parseInt(str[i]);
//System.out.print(nums[i]+" ");
}
//System.out.println();
int[] res = new Helloworld().topKFrequent(nums,k);
for (int i = 0; i < res.length; i++) {
System.out.print(res[i]+" ");
}
}
}