**Leetcode 373. 查找和最小的 K 对数字

力扣

类似第k小素分数 核心思路：pair的第二个数枚举，作为初始化，假设第二个数有m种可能，等于有m个队列，那么就等于从nums1里面逐步加入数，等价于多路合并队列 每个队列往后走一步，具体看代码

class Solution {
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int kk) {
        long long k = min((long long)kk, (long long)(nums1.size() * nums2.size()));
        // cout << "k:" << k << endl;
        // if(nums2[0] < nums1[0])swap(nums1, nums2);
        auto cmp = [&](pair<int,int>&a,pair<int,int>&b) {
            return nums1[a.first] + nums2[a.second] > nums1[b.first] + nums2[b.second];
            // return arr[a.first] * arr[b.second] > arr[b.first] * arr[a.second];
        };
        priority_queue< pair<int, int>, vector<pair<int, int>>, decltype(cmp) > q(cmp);
        vector< vector<int> >ret;
        for(int i = 0; i < nums2.size(); i++) {
            q.push(make_pair(0, i));
        }
        //  q.push(make_pair(0, 0));
        for(int i = 0; i < k; i ++) {
            auto top = q.top();
            q.pop();
            ret.push_back(
                vector<int>{nums1[top.first], nums2[top.second]}
            );
            if(top.first + 1< nums1.size()) q.push( make_pair(top.first+1,top.second) );
            // if(top.second + 1< nums2.size()) q.push( make_pair(top.first,top.second+1) );
        }
        return ret;

    }
};