Kotlin过滤列表后取最大值是否存在更优的单迭代实现方案?
单次遍历实现方案
filter + maxByOrNull的写法确实会执行两次列表遍历,同时filter操作会生成一个存储过滤结果的中间临时集合,当列表数据量较大时会产生额外的性能开销,以下是仅需单次遍历的实现版本:
- 手写循环实现(性能最优,无逻辑边界问题)
遍历过程中同步完成分类过滤、最大值比对两个动作,全程仅走一次列表,不会生成额外集合:
var maxTarget: YourObjectType? = null for (item in objects) { // 跳过需要排除的分类 if (item.category.name == xy) continue // 遇到符合要求的项时更新最大值记录 if (maxTarget == null || item.value > maxTarget.value) { maxTarget = item } }
这个实现时间复杂度为O(n),没有额外的内存开销,逻辑严谨不会出现边界异常,是性能最好的实现方式。
- 标准库fold实现(无需手写循环,逻辑严谨)
如果不想写手写for循环,可以用Kotlin标准库的fold聚合方法,同样是单次遍历完成所有逻辑,不会生成中间集合:
val maxTarget = objects.fold<YourObjectType, YourObjectType?>(null) { currentMax, item -> when { item.category.name == xy -> currentMax currentMax == null || item.value > currentMax.value -> item else -> currentMax } }
- 极简写法(存在边界限制,谨慎使用)
如果你能确定所有合法项的value永远不会等于对应数值类型的最小值,也可以直接在maxByOrNull中给需要排除的项返回一个理论最小值,实现最短代码的单次遍历逻辑:
// 风险提示:如果所有项都被排除,或是存在合法项的value刚好等于占位的最小值,返回结果会不符合预期 val maxTarget = objects.maxByOrNull { if (it.category.name == xy) Long.MIN_VALUE else it.value }
补充说明:如果处理的列表长度很小(仅几十到上百个元素),两次遍历和单次遍历的实际性能差异几乎感知不到;但如果是十万级以上的大列表,单次遍历无中间集合的实现在耗时、内存占用上的优势会非常明显。
内容的提问来源于stack exchange,提问作者David
相关产品推荐
相关产品推荐

