Kotlin函数式编程：使用fold实现列表元素统计及筛选列表的问题排查

第一个问题：count函数的错误与修正 #

你遇到的问题核心是在fold的lambda中错误使用了return。在Kotlin中，lambda里的return默认会直接返回外层的count函数，而不是把值传递给fold的累加器。所以当第一个符合条件的元素（也就是1）被处理时，函数直接返回了0+1=1，后续的元素根本没有被遍历到，这就是输出为1而非预期3的原因。

正确的count实现 #

只需要去掉lambda中的return，直接返回累加器的更新值即可——fold的lambda返回值会自动作为下一次迭代的累加器：

fun <A> count(list: List<A>, predicate: (A) -> Boolean): Int {
    return list.fold(0) { acc, a ->
        if (predicate(a)) acc + 1 else acc
    }
}

关于fold的理解确认 #

你的理解完全正确！fold确实是从指定的初始值（这里是0）开始，从左到右依次将每个元素与当前累加器传入lambda，lambda的返回值作为下一轮的累加器，最终得到一个单一的累积结果。

第二个问题：用fold返回符合条件的列表 #

你的代码报错是因为acc.add(a)返回的是Boolean（表示元素是否成功添加），但fold的lambda要求返回的是下一次迭代的累加器（也就是List<A>类型）。有两种常见的解决方式：

方式一：使用可变列表（实用派） #

在添加元素后，手动返回累加器本身即可：

fun <A> returnList(list: List<A>, predicate: (A) -> Boolean): List<A> {
    return list.fold(mutableListOf()) { acc, a ->
        if (predicate(a)) {
            acc.add(a) // 执行添加操作，忽略返回的Boolean
        }
        acc // 返回累加器，作为下一轮的acc
    }
}

这种方式性能较好，但因为修改了可变对象，不完全符合纯函数式编程的不可变原则。

方式二：使用不可变列表（纯函数式风格） #

每次符合条件时，创建一个新的列表并返回，保持数据不可变：

fun <A> returnList(list: List<A>, predicate: (A) -> Boolean): List<A> {
    return list.fold(emptyList()) { acc, a ->
        if (predicate(a)) acc + a else acc
    }
}

这里acc + a会生成一个包含原列表所有元素加上新元素a的新列表，完全遵循不可变的函数式编程理念，适合对代码风格有严格要求的场景。

内容的提问来源于stack exchange，提问作者Pythagorion