查找和最小的K对数字(LeetCode)

题目

给定两个以 非递减顺序排列 的整数数组 nums1nums2, 以及一个整数 k

定义一对值 (u,v),其中第一个元素来自 nums1,第二个元素来自 nums2

请找到和最小的 k 个数对 (u1,v1), (u2,v2) ... (uk,vk)

解题

python 复制代码
from heapq import heappush, heappop


def k_smallest_pairs(nums1, nums2, k):
    # 存储结果的列表
    result = []

    # 边界条件
    if not nums1 or not nums2:
        return result

    # 使用最小堆
    min_heap = []

    # 初始化最小堆,存储 (和, nums1中的索引, nums2中的索引)
    for i in range(min(len(nums1), k)):  # 只需要前 k 个元素进行初始化
        heappush(min_heap, (nums1[i] + nums2[0], i, 0))

    while k > 0 and min_heap:
        current_sum, i, j = heappop(min_heap)
        result.append((nums1[i], nums2[j]))

        # 如果还有下一个元素,则将下一个元素的对加到堆中
        if j + 1 < len(nums2):
            next_sum = nums1[i] + nums2[j + 1]
            heappush(min_heap, (next_sum, i, j + 1))

        k -= 1

    return result


# 测试代码
nums1 = [1, 7]
nums2 = [3, 4]
k = 3
print(f"和最小的{k}个数对 : {k_smallest_pairs(nums1, nums2, k)}")

和最小的3个数对 : [(1, 3), (1, 4), (7, 3)]

相关推荐
xiaoh_76 分钟前
解决视频处理中的 HEVC 解码错误:Could not find ref with POC xxx【已解决】
python·ffmpeg·音视频
明月与玄武36 分钟前
Python编程的真谛:超越语法,理解编程本质
python·编程语言
CodeCraft Studio38 分钟前
Excel处理控件Aspose.Cells教程:使用 Python 在 Excel 中进行数据验
开发语言·python·excel
知来者逆1 小时前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊1 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北1 小时前
力扣-160.相交链表
算法·leetcode·链表
拾忆-eleven1 小时前
C语言实战:用Pygame打造高难度水果消消乐游戏
c语言·python·pygame
草莓啵啵~1 小时前
搜索二叉树-key的搜索模型
数据结构·c++
旦莫2 小时前
Python 教程:我们可以给 Python 文件起中文名吗?
开发语言·python
豌豆花下猫2 小时前
Python 潮流周刊#99:如何在生产环境中运行 Python?(摘要)
后端·python·ai