Pergunta

Possível duplicata:
O que devo usar para uma implementação de heap máximo em Python?

Estou tentando implementar de alguma forma o heapq do python, mas para um heap máximo.Uma solução é usar (-1) e múltiplos com números da fila, mas isso não me ajuda, pois preciso armazenar URLs no heap.Então, eu quero um heapq máximo onde possa exibir o maior valor.

Foi útil?

Solução

Envolva os objetos em um wrapper de comparação reversa:

import functools

@functools.total_ordering
class ReverseCompare(object):
    def __init__(self, obj):
        self.obj = obj
    def __eq__(self, other):
        return isinstance(other, ReverseCompare) and self.obj == other.obj
    def __le__(self, other):
        return isinstance(other, ReverseCompare) and self.obj >= other.obj
    def __str__(self):
        return str(self.obj)
    def __repr__(self):
        return '%s(%r)' % (self.__class__.__name__, self.obj)

Uso:

import heapq
letters = 'axuebizjmf'
heap = map(ReverseCompare, letters)
heapq.heapify(heap)
print heapq.heappop(heap) # prints z
Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow
scroll top