【LeetCode】Heap

[215] Kth Largest Element in an Array [Medium]

给个无序数组，返回第K大的数字。

方法1. 直接使用优先队列 priority_queue

 class Solution {
 public:
     int findKthLargest(vector<int>& nums, int k) {
         priority_queue<int> pq; // priority queue 实现一个大根堆
         for (int i = ; i < nums.size(); ++i) {
             pq.emplace(nums[i]);
         }
         int i = ;
         while(i++ < k) {
             pq.pop();
         }
         return pq.top();
     }
 };

方法2. 自己实现堆排序 （周末补上）

[373] Find K Pairs with Smallest Sums [Medium]

给两个有序数组，返回前k个和最小的组合对。

思路：用一个大根堆，放k个组合对，如果新来的小于当前的最大值，就把最大的踢出去。

 /* author: wyzhang
  * 2017/03/25
  * 最大堆
 */
 class Solution {
 public:
     struct cmp {
         bool operator()(pair<int, int> a, pair<int, int> b) {
             return a.first + a.second < b.first + b.second;
         }
     };
     vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
         priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pq;
         for (size_t i = ; i < nums1.size() && i < k; ++i) {
             for (size_t j = ; j < nums2.size() && j < k; ++j) {
                 if (pq.size() < k) {
                     pq.push(make_pair(nums1[i], nums2[j]));
                 } else {
                     if (nums1[i] + nums2[j] < pq.top().first + pq.top().second) {
                         pq.pop();
                         pq.push(make_pair(nums1[i], nums2[j]));
                     }
                 }
             }
         }
         int i = min((size_t)k, pq.size());
         vector<pair<int, int>> ans(i);
         while(!pq.empty()) {
             ans[i-] = pq.top();
             pq.pop();
             i--;
         }
         return ans;
     }
 };