Scala suma częściowa z bieżącymi i wszystkimi przeszłymi elementami na liście

Question

Mamy listę takich liczb całkowitych jak: [1,4,5,6,6,7,9].

Chodzi o to, aby wygenerować listę o tej samej długości i sumach do bieżącego elementu, takiego jak: [1,5,10,16,22,29,38].

W świecie Javy to będzie wyglądać:

int sum = 0; 
int[] table = {1,4,5,6,6,7,9} 
int[] res = new int[table.length] 
for(int i=0; i<table.length; i++) { 
    sum += table[i] 
    res[i] = sum 
} 

wiem istnieją bardziej eleganckie i wydajne rozwiązania. Moje pytanie brzmi: jak zrobić coś takiego w Scali w bardziej funkcjonalny sposób?

Thx!

Answer 1

Poszukujesz kombinatora skanów.

List(1,4,5,6,6,7,9).scanLeft(0)(_ + _) 
res1: List[Int] = List(0, 1, 5, 10, 16, 22, 29, 38) 

Usuń wiodący element z ogonem, jeśli chcesz Nie jestem świadomy wersji skanowania, która nie przyjmuje wartości początkowej. Złożoność jest dla tego gościa O (n) i możesz ją zaimplementować samodzielnie, składając listę przez akumulator i listę (zawierającą poprzednie akumulatory). Ten ostatni bierzesz w rezultacie.

Answer 2

@ odpowiedź uberwatch jest słuszna, ale przez wzgląd na kompletność, tutaj jest bardziej „rodzajowe” rozwiązanie funkcjonalne używając foldLeft:

val xs = Vector(1,4,5,6,6,7,9) 
val (sumList, sum) = 
    xs.foldLeft((Vector.empty[Int], 0)) { 
    case ((list, total), x) => 
     val newTotal = x + total 
     (list :+ newTotal, newTotal) 
    } 
// sumList: Vector(1, 5, 10, 16, 22, 29, 38) 
// sum: Int = 38