IntStream 映射函数行为

Question

我有一小段代码可以计算给定数组中正值、负值和零值的数量。我可以通过迭代数组并计算正值、负值和零值的数量来使用 Java 7 轻松完成此操作。但是对于 Java 8 流，在 map 函数中编写条件是否正确？目前此代码仅迭代一次并跳过数组的其余值。

static void calculatePositiveNegativeAndZero(int[] arr) {
    AtomicInteger positive = new AtomicInteger(0);
    AtomicInteger zero = new AtomicInteger(0);
    AtomicInteger negative = new AtomicInteger(0);
    int len = arr.length;
    Arrays.stream(arr)
            .parallel()
            .map(i -> i > 0 ? positive.getAndIncrement() : (i == 0 ? zero.getAndIncrement() : negative.getAndIncrement()));

    System.out.println(positive.doubleValue()/len);
    System.out.println(negative.doubleValue()/len);
    System.out.println(zero.doubleValue()/len);
}

Answer 1

您的流管道中没有终端操作，因此根本不应该执行它。你可以用forEach代替map：

Arrays.stream(arr)
        .parallel()
        .forEach(i -> {
             int k = i > 0 ? positive.getAndIncrement() : (i == 0 ? zero.getAndIncrement() : negative.getAndIncrement());
        });

Answer 2

不用AtomicXXX 也可以做同样的事情，但方式有点不同：

Map<Integer, Long> map = Arrays.stream(arr)
            .boxed()
            .collect(Collectors.groupingBy(
                    Integer::signum,
                    Collectors.counting()));

System.out.println("positive = " + map.get(1) + " negative = " + map.get(-1) + " zero = " + map.get(0));

Answer 3

您也可以通过创建一个包装类来处理这个问题。而且您不需要使用 AtomicInteger。使用 Java 10 我通过以下方式解决了这个问题。

var wrapper = new Object(){ int positive = 0, negative = 0, zero = 0; };
Arrays.stream(arr)
            .parallel()
            .forEach(i -> { int tmp = i > 0 ? wrapper.positive++ : (i == 0 ? wrapper.zero++ : wrapper.negative++);});