在foldLeft中递增变量

Question

我有像这样的Scala代码

var i = 1
for(e <- array) {
    acc += e * i
    i += 1
}

我需要将数组中的第一个元素乘以1,将下一个元素乘以2,将下一个元素乘以3,然后将它们全部添加到累加器中.我觉得在Scala中有更好的方法,甚至可以折叠？

Answer 1

"更好"取决于你的目标是什么.简短明了？大概

{ for (i <- array.indices; e = array(i)) yield (i+1)*e }.sum

要么

array.indices.map(i => (i+1)*array(i)).sum

(或稍微快一点,因为你随时创建中间体:

array.indices.iterator.map(i => (i+1)*array(i)).sum

).

你通常应该简短明了.

快速？然后你需要去老派:

var i = 0
var acc = 0
while (i < array.length) {
  acc += (i+1)*array(i)
  i += 1
}

或使用递归

def sum(a: Array[Int], i: Int = 0, acc: Int = 0): Int =
  if (i >= a.length) acc else sum(a, i+1, (i+1)*a(i) + acc)
sum(array)

Answer 2

我更喜欢zipWithIndex,它更易于阅读:

array.zipWithIndex.map { case (e, i) => e * (i + 1) }.sum

Answer 3

val x = List(1,1,1,1,1,1)
(((0,1) /: x){case ((acc, mult), l) => (acc + (l * mult), mult + 1) })._1

换句话说,从累加器0和乘数1开始,折叠列表中的每个元素,将累加器更改为acc + (l * mult)并将乘数递增1.我们也在最后得到最终乘数,所以我们调用._1只是得到累加器.

编辑:正如@RexKerr在下面的答案(和评论)中指出的那样,如果性能是一个主要问题,那么最好使用显式递归方法.