数据结构 leetcode 373. Find K Pairs with Smallest Sums

最新推荐文章于 2025-08-23 16:01:06 发布

weixin_34183910

最新推荐文章于 2025-08-23 16:01:06 发布

阅读量52

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：https://segmentfault.com/a/1190000009675277

本文介绍了一种使用优先队列寻找两个有序整数数组中和最小的k对元素的方法。通过维护一个优先队列来保证每次都能选取当前状态下最小和的一对元素，并以此为基础探索可能的下一个最小和对。

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

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.

vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
    typedef pair<int,pair<int,int>> type;
    priority_queue<type, vector<type>, greater<type>> que;

    vector<pair<int,int>> res;

    unordered_set<long long> uset;
    que.emplace(make_pair(nums1.front() + nums2.front(), make_pair(0, 0)));

    for (int i = 0; i < k && !queue.empty(); ++i) {
        auto e = que.top();
        que.pop();
        int idx1 = e.second.first;
        int idx2 = e.second.second;
        res.emplace_back(make_pair(nums1[idx1], nums2[idx2]));

        if (idx1 + 1 < nums1.size()) {
            que.emplace(make_pair(nums1[idx1 + 1] + nums2[idx2], make_pair(idx1 + 1, idx2)));
        }
        if (idx1 == 0 && idx2 + 1 < nums2.size()) {
            que.emplace(make_pair(nums1[idx1] + nums2[idx2 + 1], make_pair(idx1, idx2 + 1)));
        }
    }
    return res;
}