题目来源
题目描述
题目解析
题目第k大的元素含义是“将数字从左到右降序排列后,从左边开始数的第k个数”。
小根堆:门槛元素就是最小的,只有大于这个门槛元素的才能进入到堆里面
因此,我们维护一个k长度的小根堆(相当于降序排序,最大的那个元素(第0大)在堆里面),那么堆顶元素就是第k大的元素
class KthLargest {
public:
KthLargest(int k, vector<int>& nums) {
qK = k;
for(auto n : nums){
qMin.push(n);
if(qMin.size() > qK){
qMin.pop();
}
}
}
int add(int val) {
qMin.push(val);
if(qMin.size() > qK){
qMin.pop();
}
return qMin.top();
}
private:
std::priority_queue<int, std::vector<int>, std::greater<int>> qMin;
int qK;
};