gpt4 book ai didi

kotlin - Kotlin 中的 Swift 枚举等效项

转载 作者:行者123 更新时间:2023-12-02 08:42:57 30 4
gpt4 key购买 nike

在 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 中找到 mapreduce 等价物,如 flatMapfold 但找不到枚举

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

最佳答案

从此输入开始:

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(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] }

关于kotlin - Kotlin 中的 Swift 枚举等效项,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52736364/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com