3

我可以为向量编写以下内容：

def add[K,V](map: Map[K,Vector[V]], key: K, values: Vector[V]): Map[K,Vector[V]] = {
  map + (key -> (map.getOrElse(key, Vector.empty) ++ values))
}

用法：

scala> add(Map(1 -> Vector(1,2,3)), 1, Vector(4,5,6))
res1: Map[Int,Vector[Int]] = Map(1 -> Vector(1, 2, 3, 4, 5, 6))

Sets 的代码几乎相同：

def add[K,V](map: Map[K,Set[V]], key: K, values: Set[V]): Map[K,Set[V]] = {
  map + (key -> (map.getOrElse(key, Set.empty) ++ values))
}

如何制作适用于所有 Iterables 的单个函数？我试着这样写：

def add[K,V](map: Map[K,Iterable[V]], key: K, values: Iterable[V]): Map[K,Iterable[V]] = {
  map + (key -> (map.getOrElse(key, Iterable.empty) ++ values))
}

但在这种情况下，我丢失了类型信息：

scala> add(Map(1 -> Set(1,2,3)), 1, Set(4,5,6))
res4: Map[Int,Iterable[Int]] = Map(1 -> Set(5, 1, 6, 2, 3, 4))

我尝试了以下方法：

def add[K,V,I[_] <: Iterable[_]](map: Map[K,I[V]], key: K, values: I[V]): Map[K,I[V]] = {
  map + (key -> (map.get(key).map(_ ++ values).getOrElse(values)))
}

但它没有编译：

无法基于 Repr 类型的集合构造带有 Any 类型元素的 That 类型的集合。[错误] map + (key -> (map.get(key).map(_ ++ values).getOrElse(values)))

4

2 回答 2

5

您需要CanBuildFrom用于通过调用来引导有关已使用集合的类型信息++。++为此，请参阅on的重载TraversableLike。

以下工作（不关心细节，如未定义的键）。

def add[K,V,It <: TraversableLike[V,It]]
  (map: Map[K,It],k: K, vs: Traversable[V])
  (implicit bf: CanBuildFrom[It,V,It])
  :Map[K,It] = 
    map + (k -> (map(k) ++ vs))
于 2013-08-23T12:56:59.340 回答
2

丑陋的 instanceof 但有效 :)

def add[K,V,I[V] <: Iterable[V]](map: Map[K,I[V]], key: K, values: I[V]): Map[K,I[V]] = {
       val m: Map[K, Iterable[Any]] =  map + (key -> (map.get(key).map(_ ++ values.asInstanceOf[Iterable[V]]).getOrElse(values)))
       m.asInstanceOf[Map[K, I[V]]]
}

来自 Scala 控制台的结果：

scala> def add[K,V,I[V] <: Iterable[V]](map: Map[K,I[V]], key: K, values: I[V]): Map[K,I[V]] = {
     |        val m: Map[K, Iterable[Any]] =  map + (key -> (map.get(key).map(_ ++ values.asInstanceOf[Iterable[V]]).getOrElse(values)))
     |        m.asInstanceOf[Map[K, I[V]]]
     | }
warning: there were 1 feature warning(s); re-run with -feature for details
add: [K, V, I[V] <: Iterable[V]](map: Map[K,I[V]], key: K, values: I[V])Map[K,I[V]]

scala> add(Map(1 -> Vector(1,2,3)), 1, Vector(4,5,6))
res0: Map[Int,scala.collection.immutable.Vector[Int]] = Map(1 -> Vector(1, 2, 3, 4, 5, 6))

scala> add(Map(1 -> Set(1,2,3)), 1, Set(4,5,6))
res1: Map[Int,scala.collection.immutable.Set[Int]] = Map(1 -> Set(5, 1, 6, 2, 3, 4))
于 2013-08-23T11:45:26.950 回答