gpt4 book ai didi

Scala 列表操作

转载 作者:行者123 更新时间:2023-12-02 15:25:09 28 4
gpt4 key购买 nike

给定一个 Int 列表和 Int 类型的变量 X。 Scala 中最好的功能方法是仅保留列表中的那些值(从列表的开头开始),以便列表值的总和小于等于变量。

最佳答案

这非常接近单行:

def takeWhileLessThan(x: Int)(l: List[Int]): List[Int] =
l.scan(0)(_ + _).tail.zip(l).takeWhile(_._1 <= x).map(_._2)

让我们把它分解成更小的部分。

首先,您使用scan 创建一个累积和列表。以下是它在一个小示例中的工作原理:

scala> List(1, 2, 3, 4).scan(0)(_ + _)
res0: List[Int] = List(0, 1, 3, 6, 10)

请注意,结果包括初始值,这就是我们在实现中采用 tail 的原因。

scala> List(1, 2, 3, 4).scan(0)(_ + _).tail
res1: List[Int] = List(1, 3, 6, 10)

现在我们将整个内容压缩到原始列表中。再次以我们的示例为例,如下所示:

scala> List(1, 2, 3, 4).scan(0)(_ + _).tail.zip(List(1, 2, 3, 4))
res2: List[(Int, Int)] = List((1,1), (3,2), (6,3), (10,4))

现在我们可以使用 takeWhile 在累积和大于我们的目标之前从这个列表中获取尽可能多的值。假设我们的示例中的目标是 5:

scala> res2.takeWhile(_._1 <= 5)
res3: List[(Int, Int)] = List((1,1), (3,2))

这几乎就是我们想要的——我们只需要去掉累积和:

scala> res2.takeWhile(_._1 <= 5).map(_._2)
res4: List[Int] = List(1, 2)

我们完成了。值得注意的是,这不是很有效,因为它计算了整个列表的累积总和等。可以通过多种方式优化实现,但就目前而言,它可能是在 Scala 中执行此操作的最简单的纯函数式方法(无论如何,在大多数情况下,性能不会成为问题)。

关于Scala 列表操作,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/30786398/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com