我有一个整数列表list1，想要得到另一个列表list2，其中包含从开头累积求和至当前索引的结果。如何使用Java 8 Stream API实现？
List<Integer> list1 = new ArrayList<>();
list1.addAll(Arrays.asList(1, 2, 3, 4));
List<Integer> list2 = new ArrayList<>();
// initialization
list2.add(list1.get(0));
for(int i=1;i<list1.size();i++) {
// increment step
list2.add(list2.get(i-1) + list1.get(i));
}
我该如何将以上的命令式风格代码转换为声明式风格？
list2 should be [1, 3, 6, 10]
流并不适合处理这种任务，因为涉及到状态 (累积部分和)。相反，您可以使用 Arrays.parallelPrefix：
Integer[] arr = list1.toArray(Integer[]::new);
Arrays.parallelPrefix(arr, Integer::sum);
List<Integer> list2 = Arrays.asList(arr);
这首先通过使用Collection.toArray将list1复制到数组中，该方法自JDK 11以来就可用。如果您尚未使用Java 11，则可以将第一行替换为传统的toArray调用：
Integer[] arr = list1.toArray(new Integer[0]);
对于每个索引：从零迭代到该索引，获取每个元素并求和
将int类型的数据装箱为Integer
收集到一个列表中
IntStream.range(0, list1.size())
.map(i -> IntStream.rangeClosed(0, i).map(list1::get).sum())
.boxed()
.collect(Collectors.toList());
list1.stream()
.collect(
Collector.of(
ArrayList::new,
(a, b) -> a.add(a.isEmpty() ? b : b + a.get(a.size() - 1)),
(a, b) -> { throw new UnsupportedOperationException(); }
)
);
O(n)（仅顺序工作）的解决方案如下，但我觉得它不是很优雅。我想这是品味问题。AtomicInteger ai = new AtomicInteger();
List<Integer> collect = list1.stream()
.map(ai::addAndGet)
.collect(Collectors.toList());
System.out.println(collect); // [1, 3, 6, 10]
O(n^2)。 - Yassin Hajaj你可以使用 sublist 方法从开头对当前索引之前的元素进行求和：
List<Integer> list = IntStream.range(0, list1.size())
.mapToObj(i -> list1.subList(0, i + 1).stream().mapToInt(Integer::intValue).sum())
.collect(Collectors.toList());
Stream.collect() 来实现这个功能：最初的回答List<Integer> list1 = Arrays.asList(1, 2, 3, 4);
List<Integer> list2 = list1.stream()
.collect(ArrayList::new, (sums, number) -> {
if (sums.isEmpty()) {
sums.add(number);
} else {
sums.add(sums.get(sums.size() - 1) + number);
}
}, (sums1, sums2) -> {
if (!sums1.isEmpty()) {
int sum = sums1.get(sums1.size() - 1);
sums2.replaceAll(num -> sum + num);
}
sums1.addAll(sums2);
});
list1.parallelStream()或list1.stream().parallel()代替list1.stream()即可。[1, 3, 6, 10]。