Flatten array in Kotlin

Question

I have a two dimensional array of Nodes that I want to flatten into a single array of all nodes using the flatten function of Kotlin arrays.

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

I then try to call the flatten function on the 2D array

nodes.flatten()

but I get an error: Type mismatch: inferred type is Array<Array<Node>> but Array<Array<out ???>> was expected. Is there another way I should be doing this?

Specify your error more precisely, use the original error message. — nhaarman
– nhaarman, Commented Jan 4, 2017 at 16:40

voddan · Accepted Answer · 2017-01-05 07:57:26Z

23

Use a more universal flatMap:

nodes.flatMap {it.asIterable()}

answered Jan 5, 2017 at 7:57

voddan

34.1k9 gold badges83 silver badges89 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

mfulton26 · Accepted Answer · 2017-01-04 17:21:06Z

10

Arrays in Kotlin are invariant so an Array<Array<Node>> is not an Array<Array<out T>> (which is the receiver type for flatten).

It looks like this will be fixed in Kotlin 1.1: Relax generic variance in Array.flatten · JetBrains/kotlin@49ea0f5.

Until Kotlin 1.1 is released you can maintain your your own version of 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
}

answered Jan 4, 2017 at 17:21

mfulton26

31.4k7 gold badges71 silver badges93 bronze badges

1 Comment

Zoe - Save the data dump Over a year ago

For the record, the same flatten method applies to lists.

Collectives™ on Stack Overflow

Flatten array in Kotlin

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related