力扣347. 前k个高频元素 java实现

菜鸟233号2025-12-04 19:46

前k个高频元素

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

示例 1：

**输入：**nums = [1,1,1,2,2,3], k = 2

输出：[1,2]

示例 2：

**输入：**nums = [1], k = 1

输出：[1]

示例 3：

**输入：**nums = [1,2,1,2,1,2,3,1,3,2], k = 2

输出：[1,2]

提示：

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

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

可以使用java中的优先队列PriorityQueue，比较器更改为较大值排在前。

具体实现如下：

java 复制代码

    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);
        }
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>((pair1, pair2) -> (pair2[1] - pair1[1])); // 将比较器变为比较值的大小，较大值排在前面
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {  // 使用entrySet直接取出map集合中的键值对
            priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
        }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            int[] temp = priorityQueue.poll();
            res[i] = temp[0];
        }
        return res;
    }

以上为记录分享用，代码较差请见谅