Java 8流：数组上的周期性求和

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 比您拥有的内核数量多，那么并行外部流就足够了。如果没有，请尝试并行化内部流。