java返回N个数中最大的前M个数

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/tangyaya8/article/details/83113523

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
思路：将前k个数装入队列，向后遍历，如果得到的数和队列首的频率进行比较，如果大于，则交换堆顶的元素和当前元素。直到遍历完成

class Solution {
        public List<Integer> topKFrequent(int[] nums, int k) {
        //对元素进行映射
        TreeMap<Integer, Integer> map = new TreeMap<>();
        Arrays.stream(nums).forEach(x -> map.put(x, !map.containsKey(x) ? 1 : map.get(x) + 1));
        PriorityQueue<Integer> pq = new PriorityQueue<>((Comparator.comparingInt(map::get)));
        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);
            }
        }
        List list = new LinkedList<Integer>();
        while (!pq.isEmpty()) {
            list.add(pq.remove());
        }
        return list;
    }
}

时间复杂度：
有n个元素:
O(NlogM)