【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 = 0; i < nums.size(); ++i) {
            pq.emplace(nums[i]);
        }
        int i = 1;
        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 = 0; i < nums1.size() && i < k; ++i) {
            for (size_t j = 0; 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-1] = pq.top();
            pq.pop();
            i--;
        }
        return ans;
    }
};