赞
踩
题目第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;
- };
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。