def sort(lists, k=300):     n = len(lists)     ans = []     heap = [(lists[i][0], i) for i in range(n)]     idx = {i: 1 for i in range(n)}     heapq.heapify(heap)     for _ in range(k):         val, i = heapq.heappop(heap)         ans.append(val)         heapq.heappush(heap, (lists[i][idx[i]], i))         idx[i] += 1          return ans 大概是这样
点赞 评论

相关推荐

11-27 14:21
同济大学 Java
卢来猴祖:给了这薪资关键拿不了几个月就给你踹了呀
点赞 评论 收藏
分享
牛客网
牛客网在线编程
牛客网题解
牛客企业服务