关注
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)))
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
11-04 19:37
桂林电子科技大学 运维工程师 点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 职场中那些令人叹为观止的八卦 #
10510次浏览 152人参与
# 你找工作想离家近 or 离家远? #
7709次浏览 155人参与
# 小红书开奖了 #
17283次浏览 105人参与
# 我的职场社死时刻 #
8639次浏览 97人参与
# 如何拒绝/反向PUA #
82656次浏览 372人参与
# 你父母给过你哪些不靠谱的职场建议? #
7635次浏览 138人参与
# 腾讯音乐秋招 #
422694次浏览 4738人参与
# 秋招吐槽大会 #
43612次浏览 406人参与
# 哪些公司开始补录了 #
8205次浏览 107人参与
# 那些年,我收到的‘奇葩’回复 #
4990次浏览 51人参与
# 职场中对你有帮助的书 #
23240次浏览 213人参与
# 你秋招最后悔的选择 #
7685次浏览 67人参与
# XX请雇我工作 #
6532次浏览 72人参与
# 秋招你经历过哪些无语的事 #
5008次浏览 58人参与
# 月薪多少能在一线城市生存 #
92356次浏览 671人参与
# 假如你的老板掉河里,你的工作能为他做什么 #
39719次浏览 402人参与
# 租房前辈的忠告 #
272827次浏览 7186人参与
# 通信硬件知识分享 #
39114次浏览 527人参与
# 你觉得机械有必要实习吗 #
66596次浏览 481人参与
# 交通银行工作体验 #
20817次浏览 69人参与
# 中科曙光工作体验 #
5158次浏览 23人参与
# 秋招疯了,看什么都像offer #
7740次浏览 94人参与