关注
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)))
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享

点赞 评论 收藏
分享
06-09 11:12
重庆移通学院 运营 不要停下啊:大二打开牛客,你有机会开卷了,卷起来
,去找课程学习,在牛客上看看大家面试笔试都需要会什么,岗位有什么需求就去学什么,努力的人就一定会有收获,这句话从来都经得起考验,像我现在大三了啥也不会,被迫强行考研,炼狱难度开局,啥也不会
,找工作没希望了,考研有丝丝机会
点赞 评论 收藏
分享
pluck*0916:刚开始都会这样滴
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 秋招什么时候开投比较合适? #
23701次浏览 318人参与
# 百度工作体验 #
223389次浏览 1972人参与
# 现代汽车前瞻技术研发急速编程挑战赛 #
27960次浏览 216人参与
# 机械人与华为的爱恨情仇 #
117155次浏览 946人参与
# 发工资后,你做的第一件事是什么 #
68154次浏览 229人参与
# 机械人集合!你是什么工程师? #
15805次浏览 89人参与
# 你觉得实习能学到东西吗 #
36307次浏览 712人参与
# 找不到好工作选择GAP真的丢人吗 #
78265次浏览 938人参与
# 我想去国央企的原因 #
59993次浏览 393人参与
# 如何准备秋招 #
20667次浏览 390人参与
# 工作中哪个瞬间让你想离职 #
25909次浏览 177人参与
# 入职第四天,心情怎么样 #
29451次浏览 417人参与
# 拼多多工作体验 #
28540次浏览 197人参与
# 多益网络求职进展汇总 #
29228次浏览 134人参与
# 快手求职进展汇总 #
547098次浏览 6001人参与
# 硬件应届生薪资是否普遍偏低? #
74083次浏览 514人参与
# 不考虑转正,实习多久合适 #
32324次浏览 145人参与
# 面试中,你被问过哪些奇葩问题? #
68553次浏览 796人参与
# 你们公司几号发工资 #
21227次浏览 140人参与
# 如果再来一次,你还会学硬件吗 #
125786次浏览 1402人参与
# 实习,不懂就问 #
46386次浏览 693人参与