【leetcode学习记录】

前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;
    }
}

未完待续。。。