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)))
点赞 评论

相关推荐

牛客刘北:如果暑期实习是27届的话,你要晚一年才会毕业,企业为什么会等你呢?要搞清时间逻辑呀!27届现在实习只能是在暑假实习,这是日常实习,不是暑期实习。所以多去投日常实习吧,暑期实习肯定不会要你的
点赞 评论 收藏
分享
Yki_:你要算时间成本呀,研究生两三年,博士三四年,加起来就五六年了,如果你本科去腾讯干五年,多领五年的年薪,加上公司内涨薪,可能到时候十五年总薪资也跟博士差不多
点赞 评论 收藏
分享
牛客网
牛客网在线编程
牛客网题解
牛客企业服务