الأكوام والأولويات واختيار أكبر k أثناء التدفق

import heapq

def largest_k(values, k):
    if k < 0:
        raise ValueError("k must be nonnegative")
    if k == 0:
        return []
    heap = []
    for value in values:
        if len(heap) < k:
            heapq.heappush(heap, value)
        elif value > heap[0]:
            heapq.heapreplace(heap, value)
    return sorted(heap, reverse=True)

print(largest_k([4, 1, 9, 9, 3], 3))
print(largest_k([4, 1], 0))

[9, 9, 4]
[]

تُعد القيم المكررة مرات ظهور مستقلة في المدخل. ولا تُستخدم heapreplace إلا بعد إثبات أن المرشح أفضل من الجذر الحالي.

import heapq
from itertools import count

arrivals = count()
heap = []
for priority, name in [(2, "Maps"), (1, "Rivers"), (2, "Atlas")]:
    heapq.heappush(heap, (priority, next(arrivals), {"name": name}))
order = []
while heap:
    _, _, payload = heapq.heappop(heap)
    order.append(payload["name"])
print(order)

['Rivers', 'Maps', 'Atlas']

تخرج الأولوية 1 قبل الأولوية 2؛ وتسبق Maps نظيرتها Atlas لأن عداد كسر التعادل الخاص بها أصغر.