作者:kanliyan_857 | 来源:互联网 | 2022-12-02 16:34
在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
我能找到map
和reduce
当量Kotlin
为flatMap
和fold
,但未能找到enumerated
.
我们如何才能实现与更高阶函数类似Kotlin
?
1> Marko Topoln..:
从这个输入开始:
val input: Array> = 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 2)提高到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] }