3

重複の可能性:
Python での最大ヒープの実装には何を使用しますか?

Python の heapq を何らかの方法で実装しようとしていますが、max-heap 用です。解決策は、(-1) と複数のキューを使用することですが、ヒープに URL を保存する必要があるため、これは役に立ちません。したがって、最大値をポップできる最大ヒープが必要です。

4

1 に答える 1

11

オブジェクトを逆比較ラッパーでラップします。

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
于 2012-10-01T22:43:38.213 に答える