Java 8 流 : periodic sum on array

Question

我想对数组进行周期性求和，即对索引模 n 求和。

int size=100;
double[] doubleArr = new double[size];
for (int i = 0; i < size; i++){
    doubleArr[i]=Math.random();
}
int n=2;
double[] results= new double[n];
for (int i = 0; i < doubleArr.length; i++) {
    results[i % n] += doubleArr[i];
}
System.out.println(Arrays.toString(results));

这段代码可以工作，但是由于我的数组非常大，我想并行化该操作。

Java 8 流似乎是一个很好的解决方案。

double[] results2= new double[n];
IntStream.range(0, doubleArr.length).forEach(i -> results2[i % n] += doubleArr[i]);
System.out.println(Arrays.toString(results2));

我得到了相同的结果。很好。

但是如果我想要并行化，它就会失败。

double[] results3= new double[n];
IntStream.range(0, doubleArr.length).parallel().forEach(i ->     results3[i % n] += doubleArr[i]);
System.out.println(Arrays.toString(results3));

我明白这是因为 results3 是可变的。

我想我应该通过收集在流操作内生成结果，但我不知道如何继续。

Answer 1

您可以从可能的模数范围 (0..n) 开始:

double[] result = IntStream.range(0, n)
         .mapToDouble(i -> IntStream.rangeClosed(0, (size-i-1)/n)
                                    .mapToDouble(j -> doubleArr[j*n+i]).sum())
         .toArray();

如果n大于您拥有的核心数量，那么并行外部流就足够了。如果没有，请尝试并行化内部流。