如何使用另一个列表的 id 过滤列表？

Question

我有一个ids列表。 我想过滤我的列表，只保留该列表中与 id 匹配的值。

fun filterHelper(ids: List<Int>, list: List<People>) {
     list.filter { ids.contains(it.id) }
}

但这是非常低效的。 它本质上是遍历列表 O(n^2)。 Kotlin 让我做得更好吗？

Answer 1

我最近问了一个关于切片地图的类似问题。 答案是没有好的内置 function，但是您可以通过使用Set而不是List作为您的 id 来解决问题，这使您的比较查找时间为 O(1)，所以总共 O(n) .

data class People(val id: Int)

fun main() {
    val people = listOf(People(1), People(2), People(3), People(4))
    val ids = setOf(2, 4)

    val filtered = people.filter { it.id in ids }
    println(filtered)
}

Output：

[People(id=2), People(id=4)]

---

值得一提的是，如果您已经有一个列表，您可以轻松地转换为一个集合：

list.toSet()