【LeetCode】查找和最小的K对数字
  k5pkyj2ahOZP 2023年11月02日 55 0


1.题目

【LeetCode】查找和最小的K对数字_最小堆

2. 分析

要善于使用现有的方法和库函数。

3. 代码

import heapq # 引入最小堆
class Solution:
def kSmallestPairs(self, nums1: List[int], nums2: List[int], k: int) -> List[List[int]]:
heap = []
for i in range(len(nums2)):
a = (0,i)
cur_s = nums1[0] + nums2[i]
heapq.heappush(heap, (cur_s,a))
res = []
cnt = 0
# 遍历
while(len(heap) and cnt < k):
head = heapq.heappop(heap)
cur_s,a = head
i,j = a
print(i,j)
if i+1 < len(nums1):
heapq.heappush(heap,(nums1[i+1]+nums2[j],(i+1,j)))
cnt += 1
res.append([nums1[i],nums2[j]])
return


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
  C0RG5J2hUEP5   2023年11月02日   38   0   0 c语言递归迭代c++算法
k5pkyj2ahOZP
最新推荐 更多