查找和最小的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)]

相关推荐
肥猪猪爸23 分钟前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
linux_carlos24 分钟前
环形缓冲区
数据结构
readmancynn35 分钟前
二分基本实现
数据结构·算法
萝卜兽编程37 分钟前
优先级队列
c++·算法
Bucai_不才39 分钟前
【数据结构】树——链式存储二叉树的基础
数据结构·二叉树
盼海1 小时前
排序算法(四)--快速排序
数据结构·算法·排序算法
LZXCyrus1 小时前
【杂记】vLLM如何指定GPU单卡/多卡离线推理
人工智能·经验分享·python·深度学习·语言模型·llm·vllm
Enougme1 小时前
Appium常用的使用方法(一)
python·appium
一直学习永不止步1 小时前
LeetCode题练习与总结:最长回文串--409
java·数据结构·算法·leetcode·字符串·贪心·哈希表
懷淰メ1 小时前
PyQt飞机大战游戏(附下载地址)
开发语言·python·qt·游戏·pyqt·游戏开发·pyqt5