具有自定义排序的优先队列

Question

我有这个案例类：

case class Offer(id: Int, amount: Int, interestRate: Double) extends Ordered[Offer] {

  def compare(that: Offer) = interestRate.compareTo(that.interestRate)
}

如您所见，我定义了基于Offer.interestRate. 我希望订单增加。

我创建了这些优惠：

Offer(1, 5, 4.0)
Offer(2, 5, 0.5)
Offer(3, 5, 1.5)

并将它们添加到优先级队列中：

val currentOffers: mutable.PriorityQueue[Offer] = mutable.PriorityQueue.empty[Offer]

问题是，当我这样做时，currentOffers.dequeue()我得到Offer(1, 5, 4.0).

相反，我想得到：

Offer(2, 5, 0.5)

我需要改变什么？

Answer 1

正如其他人在没有太多解释的情况下暗示的那样，问题在于您的比较功能：

def compare(that: Offer) = this.interestRate.compareTo(that.interestRate)

这匹配数字的自然排序，其中 -1 < 0 < 1。

当您在 Scala 中创建优先级队列时，它使用Ordering从Order您定义的隐式派生的。您缺少的是优先级队列会将“较高”值视为具有最高优先级（基本上它按降序排序）。

对此最简单的解决方法是将compare函数更改为其逆函数：

def compare(that: Offer) = that.interestRate.compareTo(this.interestRate)

需要注意的反转that和this。

另一种选择是Ordering在构建队列时提供：

val q = mutable.PriorityQueue(
  Offer(1, 5, 4.0),
  Offer(2, 5, 0.5),
  Offer(3, 5, 1.5)
)(Ordering.by[Offer, Double](_.interestRate).reverse)

它的作用是：“通过反向排序利率来创建优惠的排序”。

我更喜欢第二个选项，因为它提供了更好的粒度并允许在必要时使用多个排序。