题目
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
题解
先用map统计数组出现频率,然后构造一个小顶堆,当小顶堆的元素个数小于k时,就一直加入小顶堆。当小顶堆元素个数等于k时,若堆顶最小元素的频率也大于即将加入的,则舍弃该数。否则弹出栈顶元素,加入该数。代码如下
public int[] topKFrequent(int[] nums, int k) {
int n = nums.length;
if (nums == null || n == 0 || k < 1 || k > n) return new int[0];
int[] res = new int[k];
Map<Integer, Integer> map = new HashMap<>();
for (int num : nums) {
map.put(num, map.getOrDefault(num, 0) + 1);
}
PriorityQueue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>() {
public int compare(int[] m, int[] n) {
return m[1] - n [1];
}
});
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
int num = entry.getKey();
int count = entry.getValue();
if (pq.size() == k) {
if (pq.peek()[1] < count) {
pq.poll();
pq.offer(new int[]{num, count});
}
} else {
pq.offer(new int[]{num, count});
}
}
for (int i = 0; i < k; i++) {
res[i] = pq.poll()[0];
}
return res;
}