如何从集合中返回多个元素？

Question

在列表中,我需要用这个元素和所有先前元素的总和替换每个元素.第一个元素不需要更改.示例:必须转换列表(1.0,2.0,3.0,4.0)(1.0,3.0,6.0,10.0).我正在寻找最简洁和正确的方法.

我谷歌上搜索了很长时间,无法找到任何有关元素转换的有用信息.另外,我在Kotlinlang.org的标准库中找不到所需的功能.请帮忙解决这个问题.

fun accumulate(list: MutableList<Double>): MutableList<Double> {
    if (list.isEmpty()) return list
    else for (i in 0 until list.size) {
        if (i == list.indexOf(list.first())) list[i] = list.first()
        else list[i] = list.sumByDouble { it } // here's my problem
    }
    return list
}

Answer 1

您必须使用存储运行总和的变量.

fun accumulate(list: MutableList<Double>): MutableList<Double> {
    var runningSum = 0.0
    list.indices.forEach { i ->
        runningSum += list[i]
        list[i] = runningSum
    }
    return list
}

请注意,空列表不是此代码的特例.

如果您更喜欢FP方式并且非破坏性地转换列表,您可以这样写:

fun accumulate(list: List<Double>): List<Double> {
    var runningSum = 0.0
    return list.map {
        runningSum += it
        runningSum
    }
}