如何创建一个接受任何Seq [T,Int]并返回Seq原始类型的函数。

时间:2013-01-12 22:36:41

标签: scala

我很难定义这个问题。我想做这样的事情:

val list = List(("a",1), ("b",2))

private def shiftIndex[C <: Seq[_]](seq: C, amount: Int): C = {
  seq.map({ case (value, integer) => (value, integer + amount) })
}

shiftIndex(list,3) 

应该返回

List(("a",4),("b",5)): List[String,Integer]

但它应该适用于最常见的情况:

  1. seq实现了map(我知道它不是:Seq)
  2. seq存储任何东西和Int的元组。
  3. 任何资源链接都可以理解这些概念。

1 个答案:

答案 0 :(得分:5)

import scala.collection.generic.CanBuildFrom
import scala.collection.GenTraversable
import scala.collection.GenTraversableLike

def shiftIndex[T, Repr](seq: GenTraversableLike[(T, Int), Repr], amount: Int)(
    implicit bf: CanBuildFrom[Repr, (T, Int), Repr]) = {
  seq.map { case (value, integer) => (value, integer + amount) }
}

shiftIndex(List(("a", 1), ("b", 2)), 3)     // List((a,4), (b,5))
shiftIndex(Set(("a", 1), ("b", 2)), 3)      // Set((a,4), (b,5))
shiftIndex(Map(("a", 1), ("b", 2)), 3)      // Map(a -> 4, b -> 5)

虽然我可能会推荐一种更好看,更通用的解决方案。在这里,我富集 GenTraversable,以便对的所有可遍历对象都有一个方法mapVals,该方法将一个函数应用于该对的第二部分,并且返回类型与对象运行于:

class EnrichedWithMapVals[T, U, Repr](seq: GenTraversableLike[(T, U), Repr]) {
  def mapVals[R, That](f: U => R)(implicit bf: CanBuildFrom[Repr, (T, R), That]) = {
    seq.map { case (x, y) => (x, f(y)) }
  }
}
implicit def enrichWithMapVals[T, U, Repr](self: GenTraversableLike[(T, U), Repr]) = new EnrichedWithMapVals(self)

List(("a", 1), ("b", 2)).mapVals(_ + 3)     // List((a,4), (b,5))
Set(("a", 1), ("b", 2)).mapVals(_ + 3)      // Set((a,4), (b,5))
Map(("a", 1), ("b", 2)).mapVals(_ + 3)      // Map(a -> 4, b -> 5)