list - Kotlin 中的函数式编程 : Counting elements in list by using fold

Question

我的任务是使用 fold(函数式编程)编写一个函数来计算列表中满足谓词的元素数量。我得到了函数签名 fun <A> count(list<A>, predicate: (A) -> Boolean): Int . fold不仅要作为迭代使用，还要产生返回值。所以我尝试这样做:

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

            return acc
        }
    }

我写了一个 println 来检查它是否有效:

println(count3(listOf (1, -2, 3, 10, -5, 8, 12), { it > 0 && it < 10 }))

但是，我在控制台上得到的结果是1，而不是3，不知道是哪里出了问题。那么，有没有人知道我的错误在哪里或者我该如何实现该功能？

需要说明的是:Fold 累加一个值，从初始值(在本例中为 0)开始，从左到右对当前累加器和每个元素应用操作，还是我弄错了？

编辑(我希望可以编辑问题而不是提出新问题):

是否可以返回整个列表而不只是一个整数？我刚刚找到返回整数或 bool 值的示例。我尝试了什么:我使用了上面相同的函数签名。但是我不想返回一个 Int，而是想返回一个列表:

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

我发现的问题是 acc.add(a)返回 bool 值而不是列表，因此 IDE 将其标记为错误。那么有没有办法返回一个列表呢？

提前致谢。

Answer 1

通过说 return 你返回整个 count 函数。您可以使用 return@fold 代替。好喜欢

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

        return@fold  acc
    }
}

或者，也许更好的做法是这样做

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

lambda 中的最后一个表达式隐含地也是它的返回值