347.前K个高频元素

---

347.前K个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

LeetCode
https://leetcode-cn.com/problems/top-k-frequent-elements/

现用 HashMap计数，再用PriorityQueue

public int[] topKFrequent(int[] nums, int k) {
    HashMap<Integer,Integer> map = new HashMap<>();

    for(int v: nums)
        map.put(v,map.getOrDefault(v,0)+1);

    PriorityQueue<int[]> queue = new PriorityQueue<int[]>((o1, o2) -> -(o1[1]-o2[1]));
    for(Integer key : map.keySet())
        queue.add(new int[]{key,map.get(key)});

    int[] ans = new int[k];
    for(int i = 0;i < k;i++){
        ans[i] = queue.poll()[0];
    }

    return ans;
}