Kotlin合并N个列表并按索引求和的优化实现方案问询

嘿，这个需求我经常碰到！其实Kotlin里有比你当前实现更简洁、可读性更强的链式写法，而且性能也能打。先直接上最优解，再给你拆解细节：

通用版（支持不同长度的列表） #

这个方案会自动把短列表补0，和你原来的padStart效果一致，但代码更紧凑：

val combinedList = myNestedLists.run {
    // 先拿到所有列表里的最大长度，空列表的话返回0
    val maxLength = maxOfOrNull { it.size } ?: 0
    // 用fold累加每个列表的元素
    fold(List(maxLength) { 0.0 }) { accumulatedSum, currentList ->
        accumulatedSum.mapIndexed { index, total ->
            total + currentList.getOrElse(index) { 0.0 }
        }
    }
}

为什么这个写法更好？ #

• 链式清晰：用run把列表作为上下文，把长度计算和累加逻辑串在一起，没有零散的临时变量。
• 不可变优先：避免了手动创建可变列表并修改的操作，用fold和mapIndexed这种函数式操作，代码更安全、易维护。
• 自动补0：getOrElse(index) { 0.0 }替代了padStart，不用提前把所有列表都补全长度，节省了一次全量的列表拷贝。

简化版（仅当所有列表长度相同时） #

如果能保证所有输入列表的长度完全一致，还能再简化一步，用reduce替代fold：

val combinedList = myNestedLists.reduce { acc, list ->
    acc.zip(list).map { (a, b) -> a + b }
}

这个写法利用zip把两个列表的对应元素配对，再求和，最后用reduce把所有列表依次合并，代码非常简洁，但注意它会截断到最短列表的长度，所以只适合长度统一的场景。

性能对比 #

两种方案的时间复杂度都是O(n*m)（n是列表数量，m是最大长度），和你原来的实现一致，但函数式写法避免了可变列表的频繁修改，在Kotlin的标准库优化下，实际运行效率不会比手动操作差，甚至更稳定。

内容的提问来源于stack exchange，提问作者0xMatthewGroves