我经常需
要做类似的事情
coll.groupBy(f(_)).mapValues(_.foldLeft(x)(g(_,_)))
实现相同效果的最佳方法是什么,但避免使用groupBy显式构建中间集合?
您可以在包含中间结果的地图上折叠初始集合:
def groupFold[A,B,X](as: Iterable[A],f: A => B,init: X,g: (X,A) => X): Map[B,X] =
as.foldLeft(Map[B,X]().withDefaultValue(init)){
case (m,a) => {
val key = f(a)
m.updated(key,g(m(key),a))
}
}
你说收藏和我写了Iterable,但你必须考虑在你的问题中顺序是否重要.
如果你想要高效的代码,你可能会像在Rex的回答中那样使用可变映射.