leetcode Find K Pairs with Smallest Sums

不得不說，用慣了python之後 c++感到非常非常的陌生。打個最大堆居然都會遇到各種bug

You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k.

Define a pair (u,v) which consists of one element from the first array and one element from the second array.

Find the k pairs (u1,v1),(u2,v2) …(uk,vk) with the smallest sums.
看了測例就很好理解

Given nums1 = [1,7,11], nums2 = [2,4,6],  k = 3

Return: [1,2],[1,4],[1,6]

The first 3 pairs are returned from the sequence:
[1,2],[1,4],[1,6],[7,2],[7,4],[11,2],[7,6],[11,4],[11,6]

主要的做題思路就是用一個大小爲k 的最大堆，先把它填滿，然後用num1[i] + num2[j] 和top 比較，如果小就替換。

class Solution {
public:
    struct com {
        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>& n1, vector<int>& n2, int k) {
        priority_queue< pair<int, int>, vector<pair<int,int> > , com> pq;
        for (int i = 0; i < min((int)n1.size(), k); i++) {
            for (int j = 0; j < min((int)n2.size(), k); j++) {
                if (pq.size() < k) {
                    pq.push(make_pair(n1[i], n2[j]));
                } else if ((n1[i] + n2[j]) < (pq.top().first + pq.top().second)) {
                    pq.pop();
                    pq.push(make_pair(n1[i], n2[j]));
                }
            }
        }
        vector<pair<int,int> > res(pq.size());
        int i = pq.size() - 1;
        while(!pq.empty()) {
            res[i--] = pq.top();
            pq.pop();
        }
        return res;
    }
};

代碼其實不長，看下應該就很好理解了。
主要mark 下函數重載，priority_queue 的知識。
因爲priority_queue 是按照數據的優先級來進行排序的。每次push 和 pop 都會按照按照數據大小從大到小進行排序。

priority_queue<int, vector<int>, greater<int>, 自定義函數 >
我們可以通過自定義函數來規定數據的優先級

一開始我是想用重載 < 實現的，結果發現居然不行。原因是std 裏面自帶pair 的<實現，所以不會發現和調用我自己定義的重載，改成operate() 就可以了。