Kotlin的reduce()函数有不同的类型

时间:2018-06-16 10:56:55

标签: functional-programming kotlin

我正在查看数组扩展函数,发现reduce()一个

inline fun <S, T: S> Array<out T>.reduce(operation: (acc: S, T) -> S): S {
    if (isEmpty())
        throw UnsupportedOperationException("Empty array can't be reduced.")
    var accumulator: S = this[0]
    for (index in 1..lastIndex) {
        accumulator = operation(accumulator, this[index])
    }
    return accumulator
}

此处accumulator类型的S变量已分配给类型为T的数组中的第一个元素。

不能用reduce()函数的实际用例包围两种数据类型。这里的合成例子实际上没有任何意义。

open class A(var width: Int = 0)
class B(width: Int) : A(width)

val array = arrayOf(A(7), A(4), A(1), A(4), A(3))
val res = array.reduce { acc, s -> B(acc.width + s.width) }

似乎这个函数的大多数现实生活用例都使用这个签名:

inline fun <T> Array<out T>.reduce(operation: (acc: T, T) -> T): T

您能否提供一些示例,其中reduce()函数可用于不同类型。

1 个答案:

答案 0 :(得分:2)

以下是一个例子:

interface Expr {
    val value: Int
}

class Single(override val value: Int): Expr

class Sum(val a: Expr, val b: Expr): Expr {
    override val value: Int
        get() = a.value + b.value
}

fun main(args: Array<String>) {
    val arr = arrayOf(Single(1), Single(2), Single(3));
    val result = arr.reduce<Expr, Single> { a, b -> Sum(a, b) }
    println(result.value)
}