关注
import heapq def kLagestPairs(nums1, nums2, k):
ans = []
m, n = len(nums1), len(nums2)
k = min(k, m*n)
h = [(-(nums1[0]+nums2[0]), 0, 0)] if k > 0 else [] while len(ans) < k:
_, i, j = heapq.heappop(h)
ans.append(nums1[i]+nums2[j]) if i + 1 < m and j==0:
heapq.heappush(h, (-(nums1[i+1] + nums2[j]), i+1, j)) if j + 1 < n:
heapq.heappush(h, (-(nums1[i] + nums2[j+1]), i, j+1)) return ans
lst=raw_input().strip().split('-')
nums1=map(int,lst[0].split(','))
nums2=map(int,lst[1].split(':')[0].split(','))
K=int(lst[1].split(':')[1])
ans=kLagestPairs(sorted(nums1,reverse=True),sorted(nums2,reverse=True),K) print(','.join(map(str,ans)))
查看原帖
点赞 评论
相关推荐
牛客热帖
正在热议
# 25届秋招总结 #
242288次浏览 1976人参与
# 学历or实习经历,哪个更重要 #
40528次浏览 294人参与
# 北方华创开奖 #
22476次浏览 256人参与
# 地方国企笔面经互助 #
2467次浏览 6人参与
# 你最想要的公司福利是? #
39162次浏览 109人参与
# 选完offer后,你后悔学本专业吗 #
9634次浏览 73人参与
# 面试题刺客退退退 #
136660次浏览 2087人参与
# 应届生被毁约被毁意向了怎么办 #
26575次浏览 237人参与
# 查收我的offer竞争力报告 #
16232次浏览 222人参与
# 机械应届生薪资要多少才合适? #
12324次浏览 59人参与
# 一觉醒来,我觉醒了超级打工人系统 #
2761次浏览 34人参与
# 没有实习经历,还有机会进大厂吗 #
804540次浏览 13806人参与
# 你觉得第一学历对求职有影响吗? #
14803次浏览 121人参与
# 我的工作日记 #
21014次浏览 270人参与
# 不给转正的实习,你还去吗 #
1515975次浏览 16962人参与
# 寒假躺平还是提前实习 #
58035次浏览 430人参与
# 总结:哪家公司面试体验感最差 #
25433次浏览 128人参与
# 秋招OC许愿 #
226005次浏览 1864人参与
# 秋招被确诊为…… #
53957次浏览 302人参与
# 如何写一份好简历 #
600800次浏览 8413人参与
# 今年形式下双非本找得到工作吗 #
43809次浏览 414人参与