解題思路
覺得這題跟 703. Kth Largest Element in a Stream 非常像,甚至還比這題簡單,但兩個題目的難度標示卻不是這樣,不懂。
程式碼
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int, vector<int>, greater<int>> pq;
for(int i=0; i<nums.size(); i++)
{
pq.push(nums[i]);
if(pq.size() > k)
pq.pop();
}
return pq.top();
}
};
沒有留言:
張貼留言