Mapear para uma soma em Java 8

Toby:

Se eu tenho uma coleção:

List<Long> numbers = asList(2, 2, 4, 5);

Como posso mapear / processar-los para construir um total de execução. Para produzir algo como:

List<Long> runningTotals = asList(2, 4, 8, 13);

Ainda melhor, como posso construir uma lista de algo (como uma tupla) para que eu possa preservar os orignals:

((2 -> 2), (2 -> 4), (4 -> 8), (5 -> 13));
Samuel Philipp:

Atualizar : Como Holger apontou nos comentários, utilizando Stream.reduce()para esse fim não é correto. Veja Redução e mutável Redução ou Java 8 Streams - Recolher vs reduzir para mais informações.

Você pode usar Java Stream.collect()em vez de gerar a sua lista com somas:

List<Long> numbers = Arrays.asList(2L, 2L, 4L, 5L);
List<Pair> results = numbers.stream()
        .collect(ArrayList::new, (sums, number) -> {
            if (sums.isEmpty()) {
                sums.add(new Pair(number, number));
            } else {
                sums.add(new Pair(number, number + sums.get(sums.size() - 1).getSum()));
            }
        }, (sums1, sums2) -> {
            if (!sums1.isEmpty()) {
                long sum = sums1.get(sums1.size() - 1).getSum();
                sums2.forEach(p -> p.setSum(p.getSum() + sum));
            }
            sums1.addAll(sums2);
        });

Isto combina todos os números e cria um par para cada número com a adição à soma anterior. Usa a seguinte Pairclasse como auxiliar:

public class Pair {
    private long number;
    private long sum;

    public Pair(long number, long sum) {
        this.number = number;
        this.sum = sum;
    }

    public long getNumber() {
        return number;
    }

    public void setSum(long sum) {
        this.sum = sum;
    }

    public long getSum() {
        return sum;
    }
}

Você pode facilmente mudar essa classe auxiliar, se você quiser adicionar mais algumas informações.

O resultado no final é:

[
    Pair{number=2, sum=2}, 
    Pair{number=2, sum=4}, 
    Pair{number=4, sum=8}, 
    Pair{number=5, sum=13}
]

Acho que você gosta

Origin http://43.154.161.224:23101/article/api/json?id=167399&siteId=1
Recomendado
Clasificación