Zwiększanie zmiennej w foldLeft

Question

Mam kodu Scala jak to

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

muszę mnożenie pierwszego elementu w tablicy 1, następny przez 2, następnego dnia 3 i tak dalej, dodając je wszystkie w akumulator. Czuję, że jest lepszy sposób na zrobienie tego w Scali, może nawet ze złożeniem?

Answer 1

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

Innymi słowy, zaczynając od akumulatora 0 i mnożnik 1, złóż każdy element w liście, zmieniając akumulator do acc + (l * mult) i zwiększając mnożnik o 1. Otrzymujemy ostateczny mnożnik się na koniec również, więc nazywamy ._1, aby uzyskać akumulator.

Edytuj: Jako że @RexKerr wskazuje w swojej odpowiedzi poniżej (i komentarz), jeśli wydajność jest poważnym problemem, lepiej jest użyć metody jawnej rekursywnej.

Answer 2

Wolę zipWithIndex który jest prostszy do czytania:

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

Answer 3

Nie jestem pewien, czy to, co proponuję jest lepszy sposób to zrobić, czy nie, ponieważ jest bardziej funkcjonalny (== będzie wykonywać wolniej)

(0 /: (array zipWithIndex)) {(a, i) => (i._1 * (i._2 + 1)) + a}

ten wykonuje foldLeft na macierzy, która jest wytwarzana metodą zipWithIndex w http://www.scala-lang.org/api/current/index.html#scala.Array

zipWithIndex po prostu zamienia elementy kolekcji w ich indeksy.

Answer 4

"Lepsze" zależy od tego, jakie są Twoje cele. Krótki i wyraźny? Prawdopodobnie

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

lub

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

(lub nieco szybciej, ponieważ tworzenie związków pośrednich, jak przejść:

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

).

Powinieneś zazwyczaj być krótki i czysty.

Szybko? Następnie trzeba przejść starej szkoły:

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

lub stosowanie rekurencji

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)