跟着carl学算法,本系列博客仅做个人记录,建议大家都去看carl本人的博客,写的真的很好的!
代码随想录
LeetCode:347.前 K 个高频元素
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
public int[] topKFrequent(int[] nums, int k) {// 将nums数组转为map, key为对应的数字,value为出现的次数Map<Integer, Integer> map = new HashMap<>();for(int num : nums){map.put(num, map.getOrDefault(num, 0) + 1);}// 优先级队列 这里相当于小顶堆,因为要保留出现频率最高的k个,所以当总数大于k的时候 需要将堆顶元素弹出,即弹出出现频率最低的数据PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);for(Map.Entry<Integer, Integer> entry : map.entrySet()){pq.add(new int[]{entry.getKey(), entry.getValue()});if(pq.size() > k){pq.poll();}}int[] res = new int[k];for(int i = 0; i < k; i++){res[i] = pq.poll()[0];}return res;}