在 TreeMap 上映射 returns Map 而不是 Scala 中的 TreeMap

Question

我是 Scala 的新手，我正在实现一个带有多维键的 TreeMap，如下所示：

class dimSet (val d:Vector[Int])  extends IndexedSeq[Int] {
  def apply(idx:Int) = d(idx)
  def length: Int = d.length
}
… 
var vals : TreeMap[dimSet, A] = TreeMap[dimSet, A]()(orddimSet)

我有这个方法

def appOp0(t:TreeMap[dimSet,A], t1:TreeMap[dimSet,A], op:(A,A) => A, unop : (A) => A):TreeMap[dimSet,A] = {
  if (t.isEmpty) t1.map((e:Tuple2[dimSet, A]) => (e._1, unop(e._2)))
  else if (t1.isEmpty) t.map((t:Tuple2[dimSet, A]) => (t._1, unop(t._2)))
  else {
    val h = t.head
    val h1 = t1.head
    if ((h._1) == (h1._1)) appOp0(t.tail, t1.tail, op, unop) + ((h._1, op(h._2, h1._2)))
    else if (orddimSet.compare(h._1,h1._1) == 1) appOp0(t, t1.tail, op, unop) + ((h1._1, unop(h1._2)))
    else                   appOp0(t.tail, t1, op, unop) + ((h._1, unop(h._2)))
  }

}

但是 TreeMap 上的 map 方法（第二行和第三行）return是一个 Map，而不是 TreeMap

我在 repl 上尝试了一个更简单的例子，我得到了这个：

    scala> val t = TreeMap[dimSet, Double]( (new dimSet(Vector(1,1)), 5.1), (new dimSet(Vector(1,2)), 6.3), (new dimSet(Vector(3,1)), 7.1), (new dimSet(Vector(2,2)), 8.4)) (orddimSet)

scala> val tsq = t.map[(dimSet,Double), TreeMap[dimSet,Double]]((v:Tuple2[dimSet, Double]) => ((v._1, v._2 * v._2)))

<console>:41: error: Cannot construct a collection of type scala.collection.immutable.TreeMap[dimSet,Double] with elements of type (dimSet, Double) based on a collection of type scala.collection.immutable.TreeMap[dimSet,Double].
       val tsq = t.map[(dimSet,Double), TreeMap[dimSet,Double]]((v:Tuple2[dimSet, Double]) => ((v._1, v._2 * v._2)))
                                                               ^

  scala> val tsq = t.map((v:Tuple2[dimSet, Double]) => ((v._1, v._2 * v._2)))
  tsq: scala.collection.immutable.Map[dimSet,Double] = Map((1, 1) -> 26.009999999999998, (1, 2) -> 39.69, (2, 2) -> 70.56, (3, 1) -> 50.41)

我认为 CanBuildFrom 无法像其他 TreeMap 那样构建我的 TreeMap，但我找不到原因，¿我可以对 return TreeMap 做些什么？

谢谢

Answer 1

问题可能是当您调用 map 时没有可用的隐式 Ordering[dimSet]。该调用需要 CanBuildFrom，这又需要 TreeMap 键的隐式 Ordering：see in docs.

因此在调用 map 之前使 orddimSet 隐式可用：

implicit val ev = orddimSet
if (t.isEmpty) t1.map((e:Tuple2[dimSet, A]) => (e._1, unop(e._2)))

或者您可以使 Ordering[dimSet] 始终自动隐式可用，如果您在 dimSet 的伴生对象中定义隐式 Ordering：

object dimSet {
  implicit val orddimSet: Ordering[dimSet] = ??? // you code here
}

在 TreeMap 上映射 returns Map 而不是 Scala 中的 TreeMap

map on a TreeMap returns a Map and not a TreeMap in Scala

scala

treemap