Question

我有一个二维节点数组，我希望使用Kotlin数组的flatten函数将其展平为所有节点的单个数组。

    val nodes = kotlin.Array(width, { width ->
    kotlin.Array(height, { height -> Node(width, height) })
})

然后我尝试在2D数组上调用flatten函数

nodes.flatten()

但我收到错误：Type mismatch: inferred type is Array<Array<Node>> but Array<Array<out ???>> was expected。我还有另一种方法吗？

Answer 1

使用更通用的flatMap：

nodes.flatMap {it.asIterable()}

Answer 2

Kotlin中的数组为invariant，因此Array<Array<Node>>不是Array<Array<out T>>（flatten的接收者类型）。

看起来这将在Kotlin 1.1中修复：Relax generic variance in Array.flatten · JetBrains/kotlin@49ea0f5。

在Kotlin 1.1发布之前，您可以维护自己的flatten版本：

/**
 * Returns a single list of all elements from all arrays in the given array.
 */
fun <T> Array<out Array<out T>>.flatten(): List<T> {
    val result = ArrayList<T>(sumBy { it.size })
    for (element in this) {
        result.addAll(element)
    }
    return result
}

在Kotlin展平阵列

2 个答案: