优先级队列中的不同元素

Posted 2023-02-16

【中文标题】优先级队列中的不同元素

【英文标题】：Distinct elements in priority queue

【发布时间】：2019-03-10 07:46:44

【问题描述】：

我正在使用这个有界优先级队列的实现，https://gist.github.com/ryanlecompte/5746241，它运行良好。但是，我希望这个队列不包含任何具有相同排序“ord”的元素。我怎样才能做到这一点？

我尝试通过提供函数 lteq 而不是 lt 来更新 MaybeReplaceLowest 函数。

private def maybeReplaceLowest(a: A) 
    if (ord.lteq(a, head)) 
      dequeue()
      super.+=(a)
    
  

但我认为它不起作用，因为与新元素具有相同顺序的元素可能不在头部。有什么办法可以快速解决这个问题？

非常感谢。

【参考方案1】：

Scala PriorityQueue 是使用 Array 实现的。这对于查找操作非常低效，您需要为每个插入执行此操作（检查该元素是否已存在于队列中。

对于TreeSet 是查找和有序存储的理想选择。 但由于这是一个密封类（scala-2.12.8），我创建了复合类

import scala.collection.mutable

class PrioritySet[A](val maxSize:Int)(implicit ordering:Ordering[A]) 
    var set = mutable.TreeSet[A]()(ordering)

    private def removeAdditional(): this.type = 
        val additionalElements = set.size - maxSize
        if( additionalElements > 0)  set = set.drop(additionalElements) 
        this
    

    def +(elem: A): this.type = 
        set += elem
        removeAdditional()
    

    def ++=(elem: A, elements: A*) : this.type = 
        set += elem ++= elements 
        removeAdditional()
    

    override def toString = this.getClass.getName + set.mkString("(", ", ", ")")

这可以用作

>>> val set = new PrioritySet[Int](4)
>>> set ++=(1000,3,42,2,5, 1,2,4, 100)
>>> println(set)
PrioritySet(5, 42, 100, 1000)