TreeMap 上的 map 在 Scala 中返回 Map 而不是 TreeMap答案

【问题标题】：map on a TreeMap returns a Map and not a TreeMap in ScalaTreeMap 上的 map 在 Scala 中返回 Map 而不是 TreeMap
【发布时间】：2016-01-29 21:02:32
【问题描述】：

我是 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)))
  }

}

但是 TreeMaps（第二行和第三行）上的 map 方法返回的是 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，但我找不到原因，¿我该怎么做才能返回 TreeMap？

谢谢

【问题讨论】：

标签： scala treemap

【解决方案1】：

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

所以在调用map之前让orddimSet隐式可用：

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

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

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

【讨论】：