web-dev-qa-db-fra.com

API max heap intégrée dans Python

Le tas par défaut est l'implémentation de la file d'attente min et vous vous demandez s'il existe une option pour la file d'attente max? Merci.

J'ai essayé la solution en utilisant _heapify_max pour le tas max, mais comment gérer dynamiquement l'élément Push/pop? Il semble que _heapify_max ne puisse être utilisé que pendant le temps d'initialisation.

import heapq

def heapsort(iterable):
    h = []
    for value in iterable:
        heapq.heappush(h, value)
    return [heapq.heappop(h) for i in range(len(h))]

if __name__ == "__main__":

    print heapsort([1, 3, 5, 7, 9, 2, 4, 6, 8, 0])

Modifier, essayé _heapify_max ne semble pas fonctionner pour les éléments Push/pop dynamiquement. J'ai essayé les deux méthodes de sortie de la même manière, les deux sorties sont, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9].

def heapsort(iterable):
    h = []
    for value in iterable:
        heapq.heappush(h, value)
    return [heapq.heappop(h) for i in range(len(h))]

def heapsort2(iterable):
    h = []
    heapq._heapify_max(h)
    for value in iterable:
        heapq.heappush(h, value)
    return [heapq.heappop(h) for i in range(len(h))]

if __name__ == "__main__":

    print heapsort([1, 3, 5, 7, 9, 2, 4, 6, 8, 0])
    print heapsort2([1, 3, 5, 7, 9, 2, 4, 6, 8, 0])

Merci d'avance, Lin

pythonalgorithmqueuepriority-queue
23
Lin Ma

Dans le passé, j'ai simplement utilisé sortedcontainers 's SortedList pour cela, comme:

> a = SortedList()
> a.add(3)
> a.add(2)
> a.add(1)
> a.pop()
3

Ce n'est pas un tas, mais c'est rapide et fonctionne directement selon les besoins.

Si vous en avez absolument besoin pour être un tas, vous pouvez créer une classe de négation générale pour stocker vos objets.

class Neg():
    def __init__(self, x):
        self.x = x

    def __cmp__(self, other):
        return -cmp(self.x, other.x)

def maxheappush(heap, item):
    heapq.heappush(heap, Neg(item))

def maxheappop(heap):
    return heapq.heappop(heap).x

Mais cela utilisera un peu plus de mémoire.

24
U2EF1

Il y a une fonction _ heappop_max dans la dernière source cpython que vous pouvez trouver utile:

def _heappop_max(heap):
    """Maxheap version of a heappop."""
    lastelt = heap.pop()    # raises appropriate IndexError if heap is empty
    if heap:
        returnitem = heap[0]
        heap[0] = lastelt
        heapq._siftup_max(heap, 0)
        return returnitem
    return lastelt

Si vous modifiez la logique heappush à l'aide de heapq._siftdown_max vous devriez obtenir la sortie souhaitée:

def _heappush_max(heap, item):
    heap.append(item)
    heapq._siftdown_max(heap, 0, len(heap)-1)


def _heappop_max(heap):
    """Maxheap version of a heappop."""
    lastelt = heap.pop()  # raises appropriate IndexError if heap is empty
    if heap:
        returnitem = heap[0]
        heap[0] = lastelt
        heapq._siftup_max(heap, 0)
        return returnitem
    return lastelt


def heapsort2(iterable):
    h = []
    heapq._heapify_max(h)
    for value in iterable:
        _heappush_max(h, value)
    return [_heappop_max(h) for i in range(len(h))]

Production:

In [14]: heapsort2([1,3,6,2,7,9,0,4,5,8])
Out[14]: [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]

In [15]: heapsort2([7, 8, 9, 6, 4, 2, 3, 5, 1, 0])
Out[15]: [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]

In [16]: heapsort2([19,13,15,17,11,10,14,20,18])
Out[16]: [20, 19, 18, 17, 15, 14, 13, 11, 10]

In [17]: heapsort2(["foo","bar","foobar","baz"])
Out[17]: ['foobar', 'foo', 'baz', 'bar']
5
Padraic Cunningham