Find K Pairs with Smallest Sums

1. 解析

题目大意，求前k个和最小的pair对，pair中的元素一个来自第一个数组nums1，另外一个来自第二个数组nums2。

2. 分析

题目要求前k个最小的pair对，我最先想到的是，采用hashtable自动排序的功能，将来自不同数组的两个元素的和作为key，然后再将这两个元素组成的pair对作为value，两个元素的和可以相等，即意味着hashtable要能够存储key值相同的元素，所以采用multimap来存储，它与map的区别就在于可以存储key值相同的元素，将第1个数组nums1和第2个数组nums2的元素两两组合放在multimap当中，然后从其中选择前k个即可。

例如：nums1 =  [1, 2]，    nums2 = [2, 3]，   k = 2

multimap存储的值为：(3, [1, 2])、(4, [1, 3])、(4, [2, 2])、(5, [2, 3])，取前两个即可

class Solution {
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        vector<vector<int>> res;
        if (nums1.empty() || nums2.empty()) return res;
        multimap<int, vector<int>> nums; //将两者的和作为键值，相对应的数作为value
        vector<int> value;
        for (auto num1 : nums1){
            for (auto num2 : nums2){
                value.push_back(num1);
                value.push_back(num2);            
                nums.insert(pair<int, vector<int>>(num1+num2, value));
                value.clear();
            }
        }
        for (auto it : nums){
            if (res.size() == k) break;
            res.push_back(it.second);
        }
        return res;
    }
};

3. 优先队列的解法 

参考@Grandyang的解法，利用优先队列进行求解，不过我们唯一要注意的一点是，要构造比较器，因为我们要存储的元素是vector<int>类型，不是基本数据类型，它无法识别，我们需要堆顶元素存放的是pair对中元素的和为最大。每次将和最大的pair对与当前的进行比较，取较小值即可.

class Solution {
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k){
        vector<vector<int>> res;
        priority_queue< vector<int>, vector<vector<int>>, cmp> nums;
        for (int num1 : nums1){
            for (int num2 : nums2){
                if (nums.size() < k){ //先存储k个pair对
                    nums.push({num1, num2});                    
                }
                else{  //调整，每次将堆顶元素与当前元素进行比较，取较小值
                    if (num1 + num2 < nums.top()[0] + nums.top()[1]){
                        nums.pop();
                        nums.push({num1, num2});
                    }
                }
            }
        }
        while (!nums.empty()){ //得到逆序的结果
            res.push_back(nums.top());
            nums.pop();
        }
        return res;
    }
    struct cmp { //构造比较器
        bool operator() (vector<int> &a, vector<int> &b) {
            return a[0] + a[1] < b[0] + b[1];
        }
    };
};

 [1]https://www.cnblogs.com/grandyang/p/5653127.html

 [2]https://www.cnblogs.com/xzxl/p/7266404.html