如何用Python的heapq为自定义类实现带自定义比较器的大顶堆？

Python的heapq模块仅支持小顶堆，数值类对象可通过取反实现大顶堆，但自定义对象该如何操作？

比如我有一个按姓名字母顺序排序的Student类：

class Student:
    def __init__(self, name, age, grade):
        self.name = name
        self.age = age
        self.grade = grade

    def __repr__(self):
        return f"Student(name='{self.name}', age={self.age}, grade='{self.grade}')"

    def __lt__(self, other):
        return self.name < other.name

    def __eq__(self, other):
        return self.name == other.name

# 学生列表
students = [
    Student("Alice", 20, "A"),
    Student("Bob", 19, "B"),
    Student("Charlie", 21, "C"),
]

仅需以下代码即可创建基于姓名字母顺序的小顶堆：

import heapq

heapq.heapify(students)
print("小顶堆（列表形式）:")
print(students)

但如果想基于该列表创建大顶堆（按姓名字母逆序，即Charlie > Bob > Alice），又不想修改Student类的__lt__方法，有以下几种优雅的实现方案：

方案1：使用反向比较包装器 #

创建一个通用的包装类，对原对象的比较逻辑取反，无需修改原类：

class ReverseWrapper:
    def __init__(self, obj):
        self.obj = obj
    
    # 重写小于比较，实现反向逻辑
    def __lt__(self, other):
        return self.obj > other.obj
    
    # 保持原对象的字符串表示
    def __repr__(self):
        return repr(self.obj)

使用方式：

# 将所有学生对象包装后构建堆
wrapped_students = [ReverseWrapper(s) for s in students]
heapq.heapify(wrapped_students)

# 弹出元素时取包装器内的原对象
print("大顶堆弹出顺序:")
while wrapped_students:
    max_student = heapq.heappop(wrapped_students).obj
    print(max_student)

这种方式通用且灵活，适合所有自定义对象，完全不影响原类的逻辑。

方案2：基于元组的反向键存储 #

利用Python元组的逐元素比较特性，将原对象的排序键取反后与对象组合成元组，放入堆中：

max_heap = []
for student in students:
    # 将姓名的每个字符ASCII码取反作为键，原大的姓名对应更小的键
    reverse_key = tuple(-ord(c) for c in student.name)
    heapq.heappush(max_heap, (reverse_key, student))

# 弹出元素时忽略键，取原对象
print("大顶堆弹出顺序:")
while max_heap:
    _, student = heapq.heappop(max_heap)
    print(student)

这种方式无需额外定义类，但需要根据对象的排序规则手动生成反向键，适合对键逻辑清晰的场景。

方案3：利用heapq.nlargest（静态场景） #

如果只是需要一次性获取所有元素的降序排列，而非动态维护堆，可以直接使用heapq.nlargest：

# 获取按姓名逆序排列的所有学生
sorted_students = heapq.nlargest(len(students), students)
print("姓名逆序排列的学生列表:")
print(sorted_students)

注意：该方法适合静态数据，若需要频繁插入/删除元素，还是前两种方案更高效。

内容的提问来源于stack exchange，提问作者Spectacles4