可能重复:
我在 Python 中使用什么来实现最大堆?
我正在尝试以某种方式实现 python 的 heapq,但用于最大堆。一个解决方案是使用 (-1) 和多个队列编号,但这对我没有帮助,因为我需要将 url 存储在堆中。所以我想要一个最大堆,我可以在其中弹出最大值。
可能重复:
我在 Python 中使用什么来实现最大堆?
我正在尝试以某种方式实现 python 的 heapq,但用于最大堆。一个解决方案是使用 (-1) 和多个队列编号,但这对我没有帮助,因为我需要将 url 存储在堆中。所以我想要一个最大堆,我可以在其中弹出最大值。
将对象包装在反向比较包装器中:
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)
用法:
import heapq
letters = 'axuebizjmf'
heap = map(ReverseCompare, letters)
heapq.heapify(heap)
print heapq.heappop(heap) # prints z