LeetCode[703]数据流中的第K大元素

news2024/11/19 10:33:42

难度：简单

题目：

设计一个找到数据流中第 k 大元素的类（class）。注意是排序后的第 k 大元素，不是第 k 个不同的元素。

描述：

请实现 KthLargest 类：
KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。
int add(int val) 将 val 插入数据流 nums 后，返回当前数据流中第 k 大的元素。

示例：

输入：

["KthLargest", "add", "add", "add", "add", "add"]

[[3, [4, 5, 8, 2]], [3], [5], [10], [9], [4]]

输出：

[null, 4, 5, 5, 8, 8]

解释：

KthLargest kthLargest = new KthLargest(3, [4, 5, 8, 2]);

kthLargest.add(3); // return 4

kthLargest.add(5); // return 5

kthLargest.add(10); // return 5

kthLargest.add(9); // return 8

kthLargest.add(4); // return 8

提示：

1 <= k <= 104

0 <= nums.length <= 104

-104 <= nums[i] <= 104

-104 <= val <= 104

最多调用 add 方法 104 次

题目数据保证，在查找第 k 大元素时，数组中至少有 k 个元素

重点！！！解题思路

第一步：

由题意可知，add方法要求求得数据流中的第k大元素
求集合最值问题均可采用堆（优先队列）来解决
由于求的是第k大元素，所以我们采用小顶堆来解决此题

第二步：

题中给出KthLargest方法为初始化方法
所以在此方法中我们只初始化这个堆，和堆中目前给定的元素

第三步：

题中的add方法要求将val插入，并返回当前数据流中最大值
即将此值加入堆中，使堆保持在给定大小，然后返回堆顶元素

源码：

class KthLargest {
    PriorityQueue<Integer> priorityQueue;
    int k;
    public KthLargest(int k, int[] nums) {
        priorityQueue=new PriorityQueue<>();
        this.k=k;
        for (int num : nums) {
            add(num);
        }
    }
    
    public int add(int val) {
        priorityQueue.offer(val);
        if (priorityQueue.size()>k){
            priorityQueue.poll();
        }
        return priorityQueue.peek();
    }
}

运行结果：