Question

鉴于我在Java中有两个数组，A和B我想要按元素添加元素，这会产生一个sum数组。使用循环隐式执行此操作很容易，但我想知道是否有更优雅的解决方案，可能使用guava集合或构建java utils。或者也许是一种python-ish方式，通过列表推导很容易。

示例：

A   = [2,6,1,4]
B   = [2,1,4,4]
sum = [4,7,5,8]

Answer 1

你可以这样做：

private void sum() {
    int a[] = {2, 6, 1, 4};
    int b[] = {2, 1, 4, 4};

    int result[] = new int[a.length];
    Arrays.setAll(result, i -> a[i] + b[i]);
}

这将首先创建正确大小的int result[]。

然后在昨天发布的Java 8中，简单的部分来了：

您可以Arrays.setAll(int[] array, IntUnaryOperator);
作为IntUnaryOperator，你可以创建一个lambda映射索引到结果，在这里我们选择将i映射到a[i] + b[i]，这恰好产生了我们的总和。
对于非常大的数组，我们甚至可以使用Arrays.parallelSetAll

Answer 2

您可以在数组上使用java8流和操作：

//in this example a[] and b[] are same length
int[] a = ...
int[] b = ...
int[] result = new int[a.length];
IntStream.range(0, a.length)
     .forEach(i -> result[i] = a[i] + b[i]);

The answer in java8

Java元素和2个数组

2 个答案: