Kotlin按谓词拆分列表

在Kotlin中，我们可以使用 partition 函数来按照谓词（predicate）将一个列表拆分为两个新列表。这个函数返回一个 Pair 对象，其中第一个元素是符合谓词的列表，第二个元素是不符合谓词的列表。

语法

fun <T> Iterable<T>.partition(predicate: (T) -> Boolean): Pair<List<T>, List<T>>

这个函数接收一个谓词作为参数，返回一个 Pair 对象，其中第一个元素是符合谓词的列表，第二个元素是不符合谓词的列表。

使用

val numbers = listOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

val (evenNumbers, oddNumbers) = numbers.partition { it % 2 == 0 }

println("Even numbers: $evenNumbers")
// Even numbers: [2, 4, 6, 8, 10]

println("Odd numbers: $oddNumbers")
// Odd numbers: [1, 3, 5, 7, 9]

上面的例子中，我们将一个包含整数的列表按照奇偶分为两个列表。谓词 it % 2 == 0 指定了偶数要素。我们在一个变量解构中接收了返回的 Pair 对象，并将其元素作为两个独立的列表 evenNumbers 和 oddNumbers。

性能

partition 函数的时间复杂度为 O(n)，其中 n 是列表的长度。这使得它成为处理大型数据集的理想选择。

结论

在 Kotlin 中，我们可以使用 partition 函数按照谓词（predicate）将一个列表拆分为两个新列表。这个函数易于使用，而且适用于处理大型数据集。