Kotlin 中的 Swift 枚举等价物答案

【问题标题】：Swift enumerated equivalent in KotlinKotlin 中的 Swift 枚举等价物
【发布时间】：2018-10-10 08:56:58
【问题描述】：

在 3x3 矩阵表示中，我可以在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 和reduce 等价物，如flatMap 和fold，但找不到enumerated。

我们如何在Kotlin 中实现类似的高阶函数？

【问题讨论】：

标签： kotlin kotlin-extension

【解决方案1】：

从这个输入开始：

val input: Array<Array<Int>> = arrayOf(
        arrayOf(1, 2, 3),
        arrayOf(4, 5, 6),
        arrayOf(-7, 8, 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(n²) 提高到 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] }

【讨论】：

如果您有空数组，fold 也很重要。在 99% 的情况下，我经常发现自己使用 fold 而不是 reduce。

【解决方案2】：

您似乎在寻找withIndex

【讨论】：