----学习记录-----
前K个高频元素
这里用到了优先级队列,相当于一个小根堆,保证堆顶元素为最小值。
基本思想:先扫描数组,用hashmap存储每个元素的值以及其出现的次数,然后声明一个根据元素出现次数排序的优先级队列(小根堆),因为优先级队列队首为出现次数最小的元素值,所以将元素依次存入优先级队列(大小为K),当队列满时,比较大小,如果新值大于队首值,将队首出队,并将新值入队,最终优先级队列中的K个元素就是前K个高频元素。
//leetcode-347
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);
}
System.out.println(map);
PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer a, Integer b) {
return map.get(a) - map.get(b);
}
});
for(Integer key : map.keySet()){
if(pq.size() < k){
pq.add(key);
}
else if(map.get(key) > map.get(pq.peek())){
pq.remove();
pq.add(key);
}
}
int []arr = new int[k];
for(int i = 0; i < k; i++){
arr[i] = pq.remove();
}
return arr;
}
}
未完待续。。。