关注
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)))
查看原帖
点赞 评论
相关推荐
内向的柠檬精在研究求...:??????这是在玩呢?7面给挂了,非常逆天


点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 为了求职,我做过的疯狂伪装 #
6478次浏览 95人参与
# 从顶到拉给所有面过的公司评分 #
3169次浏览 53人参与
# 小红书校招直播来了 #
82390次浏览 478人参与
# 职场破冰,你们都聊什么? #
2650次浏览 50人参与
# 晒晒你的中秋福利 #
13421次浏览 83人参与
# 聊聊这家公司值得去吗 #
542932次浏览 3627人参与
# 产品每日一题 #
59574次浏览 604人参与
# 你面试被问到过哪些不会的问题? #
13181次浏览 559人参与
# 如何快速融入团队? #
31025次浏览 278人参与
# 机械笔面试考察这些知识点 #
9260次浏览 89人参与
# 你觉得早上几点上班合适? #
81226次浏览 329人参与
# 广联达求职进展汇总 #
9475次浏览 50人参与
# 电网笔面经互助 #
45323次浏览 428人参与
# 远景求职进展汇总 #
60241次浏览 317人参与
# 秋招报数:你投了多少家公司? #
19308次浏览 204人参与
# 浪潮求职进展汇总 #
15378次浏览 131人参与
# 安克创新求职进展汇总 #
47183次浏览 496人参与
# 上班摸鱼,你都在干些什么? #
4433次浏览 74人参与
# 为什么国企只招应届生 #
197264次浏览 1212人参与
# 机械人,你最希望上岸的公司是? #
180863次浏览 1893人参与
# 秋招的嫡长offer #
18362次浏览 174人参与