列表

详情


703. 数据流中的第 K 大元素

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

请实现 KthLargest 类:

 

示例:

输入:
["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

 

提示:

相似题目

数组中的第K个最大元素

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
class KthLargest { public: KthLargest(int k, vector<int>& nums) { } int add(int val) { } }; /** * Your KthLargest object will be instantiated and called as such: * KthLargest* obj = new KthLargest(k, nums); * int param_1 = obj->add(val); */

java 解法, 执行用时: 143 ms, 内存消耗: 53 MB, 提交时间: 2019-04-06 23:23:21

class KthLargest {

    private Queue<Integer> minHeap;
    private int k;

    public KthLargest(int k, int[] nums) {
        minHeap  = new PriorityQueue<>(k); //只维护K个元素,堆顶就是第K小
        this.k   = k;
        for (int i : nums)
            add(i);
    }

    // 满了弹出去最小的
    public int add(int val) {
        if ( minHeap.size() < this.k ) {
            minHeap.add(val);
        } else {
            if ( val > minHeap.peek() ) {
                minHeap.poll();
                minHeap.add(val);
            }
        }
        return minHeap.peek();
    }
}

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */

上一题