Python中优先级队列的优先级函数

Question

如何在 Python 中编写以下 Java PriorityQueue？

PriorityQueue<Integer[]> pq = new PriorityQueue<Integer[]>(11,
        new Comparator<Integer[]>() {
          public int compare(Integer[] A, Integer[] B) {
            return A[0] < B[0] ? -1 : 1;
          }
        });

我做到了

from Queue import PriorityQueue
def my_method(self):
  pq = # I got stuck here since I need to include "comparator"

我查看了许多示例，例如 Creating a python priority Queue，但它们似乎没有定义 priorty function 或 comparator 之类的。

Answer 1

一种选择是将(A.pulp, A) 作为优先级队列中的值。这样，它将使用元组比较首先比较 pulp 值。

另一种选择是在您的类上实现__cmp__（或定义__lt__ 和__eq__ 并使用functools.total_ordering）来进行必要的比较。这假设您还没有其他比较功能。