从Kotlin的数字列表中获取一对数字

Posted 2021-04-15

我有以下列表：

[-1, -1, 2, 3, 4, 5, -1, 9, 10, 11, -1, -1, 23, 24, 25]

我想要的是以下列表：

[Pair(2,5), Pair(9,11), Pair(23,25)]

我可以使用像start和end这样的临时变量进行迭代。但我正在寻找List上的转换方法，如map，filter等，以获得我的解决方案。

答案

首先，我将定义一个操作，将原始元素拆分为连续的数字子范围，不包括-1。这可以通过Sequence<T>上的操作方便地完成，例如takeWhile和dropWhile。唯一棘手的部分是创建一个迭代器并将其用于重新排列范围的每个项目：

fun <T> Sequence<T>.split(separator: T): Sequence<List<T>> {
    val originalIterator by lazy { iterator() }
    val eachSequence = Sequence { originalIterator }
        .dropWhile { it == separator }
        .takeWhile { it != separator }
    return generateSequence { eachSequence.toList().takeIf { it.isNotEmpty() } }
}

将split应用于Lista的便利功能同样容易：

fun <T> List<T>.split(separator: T) = 
    asSequence().split(separator).toList()

然后，只需获取每个结果范围的第一个和最后一个项目：

fun <T> List<T>.splitToRanges(separator: T) = 
    split(separator).map { it.first() to it.last() }

例：

val ints = listOf(-1, -1, 2, 3, 4, 5, -1, 9, 10, 11, -1, -1, 23, 24, 25)
println(ints.splitToRanges(-1)) // [(2, 5), (9, 11), (23, 25)]

完整的可运行示例：(link)