为什么 heappush 需要 3 个参数？

Question

为什么 heappush 需要 3 个参数 arr[i].data, i 和 arr[i]。为什么它以我为参数通常需要一个参数就是合并k个sorted link list

的代码

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,i,node.next))
    return head.next

Answer 1

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

import heapq

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

v = heapq.heappop(h) # Returns 1

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

Answer 2

实际上当元组作为输入时比较器函数将采用第一个属性进行比较，如果它们相同，它将转到下一个属性，依此类推如果两个值相同，则会抛出“TypeError: unorderable types: Node() < Node()”错误因此，为了避免这种情况，采用了在元组中唯一的第二个参数 i

为什么 heappush 需要 3 个参数？

Why heappush is taking 3 pararmeter?

python

heap

priority-queue

heapq