Swift列举了Kotlin中的等价物

在3×3矩阵表示中,我可以在 Swift中找到两个对角线与一个衬垫的总和,如下所示,

let array = [
   [1,2,3],[4,5,6],[-7,8,9]
]
let d1 = array.enumerated().map({ $1[$0] }).reduce(0,+)
let d2 = array.reversed().enumerated().map({ $1[$0] }).reduce(0,+)
print(d1) // prints 15
print(d2) // prints 1

我能够在Kotlin中找到map并减少等价物作为flatMap并折叠但找不到枚举.

我们怎样才能在Kotlin中实现与高阶函数相似的功能？

解决方法

从这个输入开始：

val input: Array<Array<Int>> = arrayOf(
        arrayOf(1,3),arrayOf(4,6),arrayOf(-7,9)
)

这就是我对对角线总和的说法：

val mainDiagonalSum = input.indices
        .map { input[it][it] }
        .reduce(Int::plus)
val counterDiagonalSum = input.indices
        .map { input[input.size - 1 - it][it] }
        .reduce(Int::plus)

请注意,这是对解决方案的改进,因为它不必创建反向数组.它将时间复杂度从O(n2)提高到O(n).

如果你正在处理大型矩阵,那么通过使用fold而不是reduce,将空间复杂度从O(n)减少到O(1)也是值得的.

val mainDiagonalSum = input.indices
        .fold(0) { sum,i -> sum + input[i][i] }
val counterDiagonalSum = input.indices
        .fold(0) { sum,i -> sum + input[input.size - 1 - i][i] }