如何在 Kotlin 中对列表进行分区和类型转换

Question

在 Kotlin 中我可以：

val (specificMembers, regularMembers) = members.partition {it is SpecificMember}

但据我所知，我不能这样做：

val (specificMembers as List<SpecificMember>, regularMembers) = members.partition {it is SpecificMember}

我的问题是 - 是否有一种惯用的方法来按类划分可迭代并在需要时对这些分区部分进行类型转换。

Answer 1

如果您更频繁地需要该功能，您可以根据需要重新实现实际的partition，例如：

inline fun <reified U : T, T> Iterable<T>.partitionByType(): Pair<List<U>, List<T>> {
  val first = ArrayList<U>()
  val second = ArrayList<T>()
  for (element in this) {
    if (element is U) first.add(element)
    else second.add(element)
  }
  return Pair(first, second)
}

用法类似如下：

val (specificMembers, regularMembers) = members.partitionByType<SpecificMember, Member>()
// where specificMembers : List<SpecificMember>
// and regularMembers : List<Member> for this example

请注意，通过这种方式，您还可以将第二种类型设置为更通用的类型。我把这留给你，这是否有意义。至少这种方式不需要unchecked cast。

Simon 还使用let-usage 显示了替代方案。您也可以直接将partition（没有let 和另一个Pair）的结果转换为任何合适的值，例如：

val (specificMembers, regularMembers) = members.partition {it is SpecificMember} as Pair<List<SpecificMember>, List<Member>>

Answer 2

partition 函数将返回一个Pair<List<T>, List<T>>，其中T 是Iterable 的泛型类型。您可以使用例如再次转换分区值let:

val (specificMembers, regularMembers) = lists
    .partition { it is SpecificMember }
    .let { Pair(it.first as List<SpecificMember>, it.second) }