Leetcode Find K Pairs with Smallest Sums

markpen

于 2017-03-27 09:48:13 发布

阅读量249

点赞数

CC 4.0 BY-SA版权

分类专栏：水题 leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/markpen/article/details/66968196

leetcode 同时被 2 个专栏收录

167 篇文章

订阅专栏

水题

112 篇文章

订阅专栏

本文介绍了一种算法，该算法接收两个已排序数组作为输入，并返回这两个数组中和最小的前K个数对。通过构建一个二维矩阵来保存所有可能的数对之和，并采用类似归并排序的方法来找出最小数对。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出两个排好序的数组，求出前k个和最小的数对。

思路：归并排序的归并思路，每次取最小的数对。

class Solution {
public:
    vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        vector<vector<int>> mp;
        for(int i = 0; i < nums1.size(); ++ i) {
            vector<int> t;
            for(int j = 0; j < nums2.size(); ++ j) {
                t.push_back(nums1[i] + nums2[j]);
            }
            mp.push_back(t);
        }
        
        vector<int> pos(nums1.size(), 0);
        vector<pair<int, int>> re;
        pair<int, int> nextP;
        int total = nums1.size() * nums2.size();
        while(k -- && total --) {
            int next = 0;
            int findMin = INT_MAX;
            for(int i = 0; i < pos.size(); ++ i) {
                if(pos[i] >= nums2.size()) continue;
                
                if(findMin > mp[i][pos[i]]) {
                    findMin = mp[i][pos[i]];
                    next = i;
                }
            }
            //cout << next << " " << pos[next] << endl;
            nextP.first = nums1[next];
            nextP.second = nums2[pos[next]];
            re.push_back(nextP);
            
            pos[next] ++;
        }
        
        return re;
    }
};