当前位置：首页 > news >正文

力扣-347.前K个高频元素

news 2025/6/13 20:51:33

题目描述

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

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

进阶：你所设计算法的时间复杂度必须优于 O(n log n) ，其中 n 是数组大小。

class Solution {public int[] topKFrequent(int[] nums, int k) {Map<Integer,Integer> map = new HashMap<>();//<值,出现的次数>for (int i = 0; i < nums.length; i++) {map.put(nums[i], map.getOrDefault(nums[i],0)+1);}PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {@Overridepublic int compare(Integer o1, Integer o2) {return map.get(o2) - map.get(o1);}});//存值，自定义排序规则Set<Integer> set = map.keySet();for (Integer key: set){pq.add(key);}int[] res = new int[k];for (int i = 0; i < k; i++) {res[i] = pq.poll();}return res;}
}

小结：题目思路比较简单，难点在于优先级队列的使用和Map的遍历。

优先级队列PriorityQueue

// 自然排序（升序）
PriorityQueue<Integer> pq1 = new PriorityQueue<>();
pq1.add(3);  pq1.add(1);  pq1.add(2);  // 实际存储顺序可能无序，但poll()返回最小值并删除，peek()返回最小值// 自定义降序排序 Lambda表达式
PriorityQueue<Integer> pq2 = new PriorityQueue<>((a, b) -> b - a);
pq2.add(3);  pq2.add(1);  pq2.poll();  // 返回3// 自定义排序规则
PriorityQueue<Node> queue = new PriorityQueue<>(new Comparator<Node>() {@Override public int compare(Node n1, Node n2) {// 按 priority 升序return n1.priority - n2.priority;  }
});

Map的遍历方法

// 键找值
Set<Integer> keys = map.keySet();
for (Integer key : keys) {System.out.println(key + map.get(key));
}// 键值对
Set<Map.Entry<Integer, String>> sets = map.entrySet(); //将键值对封装为Entry类型
for (Map.Entry<Integer, String> set : sets) {System.out.println(set.getKey());System.out.println(set.getValue());
}// Lambda表达式
map.forEach((k, v) -> System.out.println(k + v));