2013-03-29 38 views
6

我试图弄清楚如何编写一个功能交换函数,该函数可以在任何Traversable[_]上工作,给定一个集合和交换索引。我想出了以下情况:Enrich-My-Library对于所有遍历器

def swap[A, CC <% Traversable[A]](xs: CC, i: Int, j: Int): Traversable[A] = { 
    xs.slice(0, i) ++ 
    xs.slice(j, j+1) ++ 
    xs.slice(i+1, j) ++ 
    xs.slice(i, i+1) ++ 
    xs.slice(j+1, xs.size) 
} 

swap(List(1,2,3,4,5), 0, 4) // => List(5,2,3,4,1) 

我想知道如何使这个成为Traversable的一个隐含的扩展,使我与List(1,2,3,4,5).swap(0, 4)调用它。我能得到的最接近的是:

import language.implicitConversions 
class RichTraversable[A, B <% Traversable[A]](xs: B) { 
    def swap(i: Int, j: Int): Traversable[A] = { 
    xs.slice(0, i) ++ 
     xs.slice(j, j+1) ++ 
     xs.slice(i+1, j) ++ 
     xs.slice(i, i+1) ++ 
     xs.slice(j+1, xs.size) 
    } 
} 
implicit def richTraversable[A, B <% Traversable[A]](ys: B)(implicit b: Traversable[A]) 
    = new RichTraversable[A, B](ys) 

不幸的是,这并不完全。调用List(1,2,3,4,5).swap(0, 4)导致以下错误:

error: No implicit view available from List[Int] => Traversable[A]

我觉得我必须失去了一些东西,或者大大过于复杂的问题。有谁知道这应该如何构造?


注:这是纯学术性的,没有在生产环境中以任何方式使用。我试图更好地处理Scala的类型系统和边界。

回答

4

如果你使用Scala的2.10:

import scala.collection.generic.CanBuildFrom 
import scala.collection.TraversableLike 

implicit class TraversableWithSwap[A, Repr <: Traversable[A]](val xs: TraversableLike[A, Repr]) extends AnyVal { 
    def swap[That](i: Int, j: Int)(implicit bf: CanBuildFrom[Repr, A, That]): That = { 
    val builder = bf(xs.asInstanceOf[Repr]) 
    builder.sizeHint(xs) 
    builder ++= xs.take(i) 
    builder ++= xs.slice(j, j + 1) 
    builder ++= xs.slice(i + 1, j) 
    builder ++= xs.slice(i, i + 1) 
    builder ++= xs.drop(j + 1) 
    builder.result 
    } 
} 

(该AnyVal东西把它变成一个值类,这意味着编译器会改写为静态函数,以避免实际运行时做包装。)

如果使用的是早期版本,删除extends AnyVal和使用隐式转换功能,而不是implicit class

implicit def toTraversableWithSwap[A, Repr <: Traversable[A]](xs: TraversableLike[A, Repr]) = new TraversableWithSwap(xs) 

,并用它:

scala> Vector(1,2,3,4,5,6,7,8,9).swap(2,5) 
res0: scala.collection.immutable.Vector[Int] = Vector(1, 2, 6, 4, 5, 3, 7, 8, 9) 

注意,它实际上并没有什么意义来定义这个功能适用于所有Traversable因为一些traversables(如SetMap等)是无序的,所以交换两个元素是没有意义的。你可能真的想要为所有的Seq定义它。

另外:我们是否也可以同意将其称为“enhance-my-library”?

+0

第一:哇,这是我从未见过的建造者?第二:是的,我们可以同意从现在开始= = – KChaloux

+0

建造者是内部集合中使用的东西。例如,请参阅:https://github.com/scala/scala/blob/master/src/library/scala/collection/TraversableLike.scala#L237 – dhg

+0

嗯。我找不到它可以找到CanBuildFrom或TraversableLike。有没有我缺少的进口? – KChaloux