class Solution(object):
def kSmallestPairs(self, nums1, nums2, k):
"""
:type nums1: List[int]
:type nums2: List[int]
:type k: int
:rtype: List[List[int]]
"""
res=[]
pq=[]
def push(i,j):
if i<len(nums1) and j<len(nums2):
heapq.heappush(pq,[nums1[i]+nums2[j],i,j])
push(0,0)
while pq and len(res)<k:
_,i,j=heapq.heappop(pq)
res.append([nums1[i],nums2[j]])
push(i,j+1)
if j==0:
push(i+1,0)
return res
373. Find K Pairs with Smallest Sums
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- You are given two integer arrays nums1 and nums2 sorted i...
- Question You are given two integer arrays nums1 and nums2...
- 这题的线索很明显 如果[1, 7, 11] 和 [2, 4, 6] 的话,第一个无疑是从first in arra...
- My code: 其实和 top k frequent number 差不多的思路。也是建一个priority q...