Question

我想在scala中对相邻元素求和，我不知道如何处理最后一个元素。

所以我有一个清单：

val x = List(1,2,3,4)

我想使用索引和地图对相邻元素求和：

val size = x.indices.size
val y = x.indices.map(i => 
    if (i < size - 1)
       x(i) + x(i+1))

问题是这种方法在最后创建了一个AnyVal元素：

res1: scala.collection.immutable.IndexedSeq[AnyVal] = Vector(3, 5, 7, ())

如果我尝试对集合中的元素或其他数字方法求和，则它不起作用：

error: could not find implicit value for parameter num: Numeric[AnyVal]

我尝试使用以下方法过滤掉元素：

y diff List(Unit) or y diff List(AnyVal)

但它不起作用。

在scala中有没有更好的方法来做这种类型的相邻和而不使用foor循环？

Answer 1

对于功能更强大的解决方案，您可以使用sliding将元素分组为两个（或任意数量），然后map分组到sum。

scala> List(1, 2, 3, 4).sliding(2).map(_.sum).toList
res80: List[Int] = List(3, 5, 7)

sliding(2)将要做的是创建这样的列表的中间迭代器：

Iterator(
    List(1, 2),
    List(2, 3),
    List(3, 4)
)

因此，当我们链接map(_.sum)时，我们会将每个内部List映射到它自己的总和。 toList会将Iterator转换回List。

Answer 2

您也可以尝试模式匹配和尾递归。

import scala.annotation.tailrec
@tailrec
def f(l:List[Int],r :List[Int]=Nil):List[Int] = {
l match {
    case x :: xs :: xss => 
    f(l.tail, r :+ (x + xs))
    case _ => r
}
}



scala> f(List(1,2,3,4))
res4: List[Int] = List(3, 5, 7)

Answer 3

通过压缩两个列表进行理解，第二个第一个项目被删除，

for ( (a,b) <- x zip x.drop(1) ) yield a+b

导致

List(3, 5, 7)

如何对scala中的相邻元素求和

3 个答案: