如何在 Kotlin 数据结构中添加两个数组

问题描述

我是 Kotlin 初学者,我正在尝试欺骗 leetcode 练习,我有一个问题。

例如

val numbers = intArray(1,2,3)
val numbers2 = intArray(4,5,6)

在循环的帮助下制作,获取

val number2 = intArray(5,7,9)

不是这个val number2 = intArray(1,3,4,6)

解决方法

您可以使用 zip() 将两个数组组合成一个 List<Pair<Int,Int>>,然后使用 map() 来计算总和:

fun main() {
    val numbers = intArrayOf(1,2,3)
    val numbers2 = intArrayOf(4,5,6)
    val sum = numbers.zip(numbers2).map { (a,b) -> a + b }
    println(sum.joinToString(","))
}

输出:

5,7,9

以上可能是最 Kotlin 的做法。但是如果你真的想生成一个数组并使用循环,你可以这样做:

fun main() {
    val numbers = intArrayOf(1,6)
    val sum = IntArray(numbers.size)

    for (i in numbers.indices)
        sum[i] = numbers[i] + numbers2[i]

    println(sum.joinToString(",9

但是,您可以通过向 IntArray 构造函数提供 lambda 来直接生成数组而无需循环:

fun main() {
    val numbers = intArrayOf(1,6)
    val sum = IntArray(numbers.size) { numbers[it] + numbers2[it] }
    println(sum.joinToString(",9