秒懂：kotlin 集合操作fold vs reduce

2021-06-07 6,932 阅读2分钟

一、理解fold 和 reduce操作

直接看源码吧，别去搜别人写的blog了，这两个函数的源码比其他文档【包括官方文档】都更通俗易懂。而且可以秒懂。

完美诠释了：优秀的代码是最好的文档。

fold

public inline fun <T, R> Iterable<T>.fold(initial: R, operation: (acc: R, T) -> R): R {
    var accumulator = initial
    for (element in this) accumulator = operation(accumulator, element)
    return accumulator
}

reduce

public inline fun <S, T : S> Iterable<T>.reduce(operation: (acc: S, T) -> S): S {
    val iterator = this.iterator()
    if (!iterator.hasNext()) throw UnsupportedOperationException("Empty collection can't be reduced.")
    var accumulator: S = iterator.next()
    while (iterator.hasNext()) {
        accumulator = operation(accumulator, iterator.next())
    }
    return accumulator
}

二、fold与reduce有一个重要且容易忽略的区别：

reduce的返回值类型必须和集合的元素类型相符。

fold的返回值类型则不受约束。

三、实践：fold()和reduce()有什么用呢？

1. 如下我们通过reduce() 实现集合求和：

val numbers = listOf(1, 1, 1)
val result = numbers.reduce { a: Int, b: Int -> a + b }
println("reduceResult=$result")//result=3

2. 将集合拼接成字符串

val numbers = listOf(1, 1, 1)
val result = numbers.fold(StringBuilder()) {
  str: StringBuilder, i: Int -> str.append(i).append(" ")
}
println("foldResult=$result")

来自大叔的灵魂拷问下大家：

如果numbers集合的元素为空时，result的结果是什么？

总结

fold : 合拢，折叠

reduce : 压缩

两个函数都是对集合的遍历，只是遍历完成之后能得到一个结果。

折叠和压缩的意思，大家可以理解为，将集合折叠成一个新的对象【对象的类型，可以与集合元素的类型无关】，或者压缩成一个值【类型必须和集合元素的类型一致】。

秒懂：kotlin 集合操作fold vs reduce

一、理解fold 和 reduce操作

fold

reduce

二、fold与reduce有一个重要且容易忽略的区别：

三、实践：fold()和reduce()有什么用呢？

1. 如下我们通过reduce() 实现集合求和：

2. 将集合拼接成字符串

来自大叔的灵魂拷问下大家：

总结

更多精彩原文：