解題思路
覺得這題跟 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(); } };
沒有留言:
張貼留言