这是对Expand a Set of Sets of Strings into Cartesian Product in Scala
的跟进你想要的想法是:
val sets = Set(Set("a","b","c"), Set("1","2"), Set("S","T"))
然后回来:
Set("a&1&S", "a&1&T", "a&2&S", ..., "c&2&T")
一般解决方案是:
def combine[A](f:(A, A) => A)(xs:Iterable[Iterable[A]]) =
xs.reduceLeft { (x, y) => x.view.flatMap {a => y.map(f(a, _)) } }
使用如下:
val expanded = combine{(x:String, y:String) => x + "&" + y}(sets).toSet
理论上,应该有一种方法可以输入Set[Set[A]]
类型并返回Set[B]
。也就是说,在组合元素时转换类型。
一个示例用法是接受字符串集(如上所述)并输出其串联的长度。 f
中的combine
函数具有以下形式:
(a:Int, b:String) => a + b.length
我无法提出实施方案。有人有答案吗?
答案 0 :(得分:9)
如果你真的想让你的组合器功能进行映射,你可以使用fold
,但是Craig pointed out你需要提供一个种子值:
def combine[A, B](f: B => A => B, zero: B)(xs: Iterable[Iterable[A]]) =
xs.foldLeft(Iterable(zero)) {
(x, y) => x.view flatMap { y map f(_) }
}
您需要这样的种子值的事实来自组合器/映射器函数类型(B, A) => B
(或者,作为curried函数,B => A => B
)。显然,要映射您遇到的第一个A
,您需要提供B
。
使用Zero
类型类可以使调用者更简单:
trait Zero[T] {
def zero: T
}
object Zero {
implicit object IntHasZero extends Zero[Int] {
val zero = 0
}
// ... etc ...
}
然后combine
方法可以定义为:
def combine[A, B : Zero](f: B => A => B)(xs: Iterable[Iterable[A]]) =
xs.foldLeft(Iterable(implicitly[Zero[B]].zero)) {
(x, y) => x.view flatMap { y map f(_) }
}
用法:
combine((b: Int) => (a: String) => b + a.length)(sets)
Scalaz提供了Zero
类型类,以及许多其他功能编程的好东西。
答案 1 :(得分:7)
你遇到的问题是减少(左|右)取一个函数(A,A)=> A不允许您更改类型。你想要更像foldLeft的东西,它取(B,A)⇒B,允许你积累不同类型的输出。 folds虽然需要种子值,但这里不能是空集合。你需要将xs分成头部和尾部,将头部迭代映射为Iterable [B],然后使用映射的头部,尾部和一些函数调用foldLeft(B,A)=> B.这似乎比它的价值更麻烦,所以我只是预先做好所有的映射。
def combine[A, B](f: (B, B) => B)(g: (A) => B)(xs:Iterable[Iterable[A]]) =
xs.map(_.map(g)).reduceLeft { (x, y) => x.view.flatMap {a => y.map(f(a, _)) } }
val sets = Set(Set(1, 2, 3), Set(3, 4), Set(5, 6, 7))
val expanded = combine{(x: String, y: String) => x + "&" + y}{(i: Int) => i.toString}(sets).toSet