Leetcode703数据流中的第k大元素

Leetcode703数据流中的第k大元素

思路： 把这k个元素建一个数据结构，可以方便的读出来其中最小的，所以可以用队列中的小顶堆，或者使用set排序应该也可以。以下是优先队列小顶堆写法

class KthLargest {
    priority_queue<int,vector<int>,greater<int>> q;
    int k;
public:
    KthLargest(int k, vector<int>& nums) {
        this->k=k;
       q.push(INT_MIN);
       for(auto num:nums)
       {
           add(num);
       } 
    }
    
    int add(int val) {
        int temp=q.top();
        if(q.size()<k)
        {
            q.push(val);
        }
        else if(val>temp)
        {
            q.pop();
            q.push(val);
        }
    return q.top();  

    }
};