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 大概是这样
点赞 评论

相关推荐

牛客网
牛客网在线编程
牛客网题解
牛客企业服务