Kotlin | 高阶函数reduce()、fold()详解

在 Kotlin 中，高阶函数 reduce() 和 fold() 可以用来对一个集合中的元素进行累加操作，将集合中的所有元素合并为一个结果。

reduce() 函数：

reduce() 函数将集合中的元素依次传入给定的操作函数，操作函数的返回值作为下一次操作的输入，最终得到一个结果。如果集合为空，则抛出 NoSuchElementException 异常。

reduce() 函数的定义如下：

inline fun <T, R> Iterable<T>.reduce(
    operation: (acc: T, T) -> T
): T

其中，operation 参数是一个操作函数，用于对集合中的元素进行累加操作。该函数接受两个参数，第一个参数是累加器，第二个参数是当前元素，返回值为累加器和当前元素的组合结果。

下面是一个 reduce() 函数的示例代码：

val numbers = listOf(1, 2, 3, 4, 5)
val sum = numbers.reduce {
    
     acc, i -> acc + i }
println(sum) // 输出：15

在上面的代码中，reduce() 函数对 numbers 集合中的元素进行累加操作，初始累加器的值为第一个元素 1，依次将累加器和后面的元素传入操作函数中进行累加操作，最终得到累加结果 15。

fold() 函数：

fold() 函数与 reduce() 函数类似，不同之处在于 fold() 函数可以指定一个初始值作为累加器的初始值。如果集合为空，则返回指定的初始值。

fold() 函数的定义如下：

inline fun <T, R> Iterable<T>.fold(
    initial: R,
    operation: (acc: R, T) -> R
): R

其中，initial 参数是累加器的初始值，operation 参数是一个操作函数，用于对集合中的元素进行累加操作。该函数接受两个参数，第一个参数是累加器，第二个参数是当前元素，返回值为累加器和当前元素的组合结果。

下面是一个 fold() 函数的示例代码：

val numbers = listOf(1, 2, 3, 4, 5)
val sum = numbers.fold(0) {
    
     acc, i -> acc + i }
println(sum) // 输出：15

在上面的代码中，fold() 函数对 numbers 集合中的元素进行累加操作，初始累加器的值为 0，依次将累加器和后面的元素传入操作函数中进行累加操作，最终得到累加结果 15。

总结：

reduce() 和 fold() 函数都可以用来对集合中的元素进行累加操作，reduce() 函数没有指定初始值，fold() 函数可以指定初始值。在使用 reduce() 和 fold() 函数时，需要注意集合是否为空，以及操作函数的返回值类型。