我有一个Grid
类,它包装在Cell
对象的二维数组上。我希望这个类实现Iterable<Cell>
接口,以便在循环中使用它并直接迭代整个单元格。有没有简单的方法可以做到这一点?Kotlin 是否支持yield return
样式迭代器?我目前的解决方案非常冗长:
override fun iterator() = object : Iterator<Cell> {
val currentOuter = grid.iterator() // grid is object of Array<Array<Cell>>
var currentInner = if (currentOuter.hasNext()) currentOuter.next().iterator() else arrayOf<Cell>().iterator()
override fun next(): Cell {
if (!hasNext()) {
throw NoSuchElementException()
}
return if (currentInner.hasNext()) {
currentInner.next()
} else {
currentInner = currentOuter.next().iterator()
currentInner.next()
}
}
override fun hasNext(): Boolean {
return currentInner.hasNext() || currentOuter.hasNext()
}
}
Kotlin 是否支持生成返回样式迭代器?
是的,它确实如此,通过协程的功能。下面是一个自包含的示例:
data class Cell(val d: Int)
val grid: Array<Array<Cell>> = arrayOf(arrayOf(Cell(1), Cell(2)), arrayOf(Cell(3), Cell(4)))
fun cellSequence() = buildSequence {
grid.forEach { it.forEach { yield(it) } }
}
fun main(args: Array<String>) {
cellSequence().forEach { println(it) }
}
虽然这个特殊的问题可以通过一个flatMap
简单地解决,但所呈现的代码可以作为模板来编写任何类型的过程式代码,例如:
fun complexCellSequence() = buildSequence {
yield(Cell(-1))
if (grid.size <= 2) {
yield(Cell(2))
}
for (row in grid) {
if (row.contains(Cell(1))) {
yield(Cell(1))
} else {
yield(Cell(12))
row.forEach { yield(it) }
}
}
}
在没有协程的情况下重写这将是非常容易的。
一个非常简单的解决方案是这样的:
val grid: Array<Array<Cell>> = ...
override fun iterator() : Iterator<Cell> = grid.flatMap { it.asList() }.iterator()