力扣 347. 前 K 个高频元素

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例二：

输入: nums = [1], k = 1
输出: [1]

提示：

• 1 <= nums.length <= 105
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

进阶：你所设计算法的时间复杂度 必须 优于 O(n log n) ，其中 n 是数组大小。

思路：

选出前k的高频元素，使用小顶堆。

代码实现：

import javax.swing.undo.UndoableEdit;
import java.sql.SQLOutput;
import java.util.*;

public class Helloworld {

    public int[] topKFrequent(int[] nums, int k){
        Map map = new HashMap();
        for (int num:nums) {
            map.put(num,map.getOrDefault(num,0) + 1);
        }
        PriorityQueue pq = new PriorityQueue<>(new Comparator() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return map.get(o1) - map.get(o2); //升序，小顶堆
                //return map.get(o1) - map.get(o2); //降序，大顶堆

            }
        });
        for (Integer num:map.keySet()) {
            System.out.println(num+" "+map.get(num));
        }
        for (Integer key:map.keySet()) {
            if(pq.size() < k)
                pq.add(key);
            else if(map.get(key) > map.get(pq.peek())){
                pq.poll();
                pq.add(key);
            }
        }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = pq.poll();
        }
        return res;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String string = scanner.nextLine();
        int k = scanner.nextInt();
        String[] str = string.split(" ");
        int[] nums = new int[str.length];
        for (int i = 0; i < str.length; i++) {
            nums[i] = Integer.parseInt(str[i]);
            //System.out.print(nums[i]+"  ");
        }
        //System.out.println();
        int[] res = new Helloworld().topKFrequent(nums,k);
        for (int i = 0; i < res.length; i++) {
            System.out.print(res[i]+" ");
        }
    }
}

最后附上Java的优先队列PriorityQueue详解