kotlin - 如何有效地从Kotlin的集合(前N个)中获取N个最低值？

Question

如何有效地从Kotlin的集合(前N个)中获取N个最低值？

除了collectionOrSequence.sortedby{it.value}.take(n)还有其他方法吗？

假设我有一个包含+100500元素的集合，我需要找到10个最低的元素。恐怕sortedby会创建新的临时集合，以后只需要10个项目。

Answer 1

您可以保留n个最小元素的列表，然后按需更新它，例如

fun <T : Comparable<T>> top(n: Int, collection: Iterable<T>): List<T> {
    return collection.fold(ArrayList<T>()) { topList, candidate ->
        if (topList.size < n || candidate < topList.last()) {
            // ideally insert at the right place
            topList.add(candidate)
            topList.sort()
            // trim to size
            if (topList.size > n)
                topList.removeAt(n)
        }
        topList
    }
}

这样，您只需将列表中的当前元素与前n个元素中的最大元素进行一次比较，通常比对整个列表进行排序 https://pl.kotl.in/SyQPtDTcQ