为什么 heappush 需要 3 个参数？

如何解决为什么 heappush 需要 3 个参数？

为什么 heappush 采用 3 个参数 arr[i].data、i 和 arr[i]。 为什么将 i 作为参数 它通常需要一一参数 这是合并k个排序链表的代码

def mergeKLists(arr,K):
# code here
# return head of merged list
    heap = []
    head = tail = Node(0)
    for i in range(K):
        heapq.heappush(heap,(arr[i].data,i,arr[i]))

    while heap:
        node = heapq.heappop(heap)
        node = node[2]
        tail.next = node
        tail = tail.next
        if node.next:
            i += 1
            heapq.heappush(heap,(node.next.data,node.next))
    return head.next

解决方法

它没有。在您的代码中，您将一个元组推送到堆中，但这不是必需的。例如：

import heapq

h = [1,2,3,4]
heapq.heapify(h)

v = heapq.heappop(h) # Returns 1

heapq.heappush(h,6) # Now h = [2,4,6]

,

实际上当元组作为输入时 比较器函数将采用第一个属性进行比较，如果它们相同，它将转到下一个属性，依此类推 如果两个值相同，它将抛出“TypeError: unorderable types: Node() i