【leetcode】703 数据流中第K大元素（堆）

题目链接：https://leetcode-cn.com/problems/kth-largest-element-in-a-stream/

题目描述

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

你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器，它包含数据流中的初始元素。每次调用 KthLargest.add，返回当前数据流中第K大的元素。

示例:

int k = 3;
int[] arr = [4,5,8,2];
KthLargest kthLargest = new KthLargest(3, arr);
kthLargest.add(3);   // returns 4
kthLargest.add(5);   // returns 5
kthLargest.add(10);  // returns 5
kthLargest.add(9);   // returns 8
kthLargest.add(4);   // returns 8

说明:
你可以假设 nums 的长度≥ k-1 且k ≥ 1。

思路

1 堆

用大小为K的最小堆存储topK大元素。堆顶对第K大元素。

复杂度分析

• 时间复杂度：O（1）取第K大元素，O（logk）调整；总时间O（n logk）
• 空间复杂度：O（k）

class KthLargest {
    multiset<int, less<int>> m;
    k;
public:
    KthLargest(int k, vector<int>& nums) {
        for (int i = 0; i < nums.size(); ++i) {
            if(m.size()<k)
                m.insert(nums[i]);
            else if(nums[i] > *m.begin()){
                m.erase(m.begin());
                m.insert(nums[i]);
            }
        }
    }

    int add(int val) {
        if(m.size() < k)
            m.insert(val);
        else if(val > *m.begin()){
            m.erase(m.begin());
            m.insert(val);
        }
        return *m.begin();
    }
};

标签：kthLargest,int,KthLargest,returns,703,add,数据流,元素,leetcode
来源： https://blog.csdn.net/zjwreal/article/details/94200899