Question

我遇到了一个奇怪的情况，contains()函数似乎在Scala中的List和TreeSet之间的工作方式不同，我不知道为什么或如何解决它。

为了简洁起见，我创建了一个名为DataStructure的课程。它包含两个元素：坐标对(i, j)和Int。（它比这更复杂，但在这个MWE中，这就是它的样子）它有一个自定义比较器，将按Int排序，我已覆盖hashCode和{ {1}}，无论equals如何，包含相同坐标对(i, j)的两个元素都被视为相等。

当我将Int的实例放入DataStructure和List时，程序找到完全匹配没有问题。但是，在检查具有相同坐标对但TreeSet不同的新元素时，Int会返回List.contains而true会返回TreeSet.contains。为什么会发生这种情况，我该如何解决？

这是我的代码简化为最小的工作示例：

班级false

DataStructure

驱动程序

package foo

class DataStructure(e1: (Int, Int), e2: Int) extends Ordered[DataStructure] {
  val coord: (Int, Int) = e1
  val cost: Int = e2

  override def equals(that: Any): Boolean = {
    that match {
    case that: DataStructure => if (that.coord.hashCode() == this.coord.hashCode()) true else false
    case _ => false
  }}
  override def hashCode(): Int = this.coord.hashCode()

  def compare(that: DataStructure) = {
    if (this.cost == that.cost)
        0
    else if (this.cost > that.cost)
        -1  //reverse ordering
    else
        1
  }    
}

输出：

package runtime

import foo.DataStructure
import scala.collection.mutable.TreeSet

object Main extends App {
        val ts = TreeSet[DataStructure]()

        val a = new DataStructure((2,2), 2)
        val b = new DataStructure((2,3), 1)

        ts.add(a)
        ts.add(b)

        val list = List(a, b)

        val listRes = list.contains(a)  // true
        val listRes2 = list.contains(new DataStructure((2,2), 0)) // true
        val tsRes = ts.contains(a)  // true
        val tsRes2 = ts.contains(new DataStructure((2,2), 0))  // FALSE!

        println("list contains exact match: " + listRes)
        println("list contains match on first element: " + listRes2)
        println("TreeSet contains exact match: " + tsRes)
        println("TreeSet contains match on first element: " + tsRes2)
}

Answer 1

几乎可以肯定List.contains正在检查equals每个元素以查找匹配项，而TreeSet.contains正在走树，并使用compare查找匹配项。

您的问题是compare与您的equals不一致。我不知道你为什么这样做，但不要：

https://www.scala-lang.org/api/current/scala/math/Ordered.html

＆＃34;重要的是，Ordered [A]实例的equals方法与compare方法一致。＆＃34;

Contains（）在List和TreeSet上的行为不同

1 个答案: