使用 Java 函数式编程简化 if-else 条件

Question

对使用 Java 8 功能简化以下代码块有何建议？

int[] ans = new int[2];
list.forEach(i -> {
    if (i > 0) {
        ans[0] += 1;
    } else if (i < 0) {
        ans[1] += 1;
    }
});

附：不知道我是否应该发布这个here

Answer 1

如果您不想计算零，您的代码将尽可能简单。但是，如果您想将零计为正数，则可以将其缩短为这个。

int[] ans = new int[2];
for (int i : list) ans[i < 0 ? 1 : 0] += 1;

Answer 2

 ans[0] = (int)list.stream().filter(x -> x < 0).count();
 ans[1] = (int)list.stream().filter(x -> x > 0).count();

但我几乎不认为这是一种简化，您的解决方案已经尽可能简单了。

如果您确实想计算零，这可以简化为：

 list.forEach(x -> ++ans[x >>> 31])

Answer 3

我建议

int[] ans = new int[2];
list.forEach(i -> ans[i >>> 31] += i==0 ? 0 : 1);

其中i >>> 31 丢弃除符号位以外的所有位（即与i<0 ? 1 : 0 相同），第二个条件句柄为零。

我不说，它确实比原版更好。

Answer 4

为了区分i > 0和i < 0这两种情况，我们可以使用Stream.partition：

Map<Boolean, List<Integer>> partitioned = list.stream()
  .filter(i -> i != 0)
  .collect(Collectors.partitioningBy(i -> i > 0));
ans[0] = partitioned.get(true).size();
ans[1] = partitioned.get(false).size();

简化了吗？至少它仍然可读且易于理解。

编辑
或者正如@saka1029 建议的那样：

Map<Boolean, Long> partitioned = list.stream()
  .filter(i -> i != 0)
  .collect(Collectors.partitioningBy(i -> i > 0, Collectors.counting()));
ans[0] = partitioned.get(true);
ans[1] = partitioned.get(false);

编辑
还有一个返回所需数组的Stream 解决方案。但我认为这并不简单。所以要比较一下。

int[] ans = list.stream().filter(i -> i != 0).collect(
  () -> new int[2], 
  (arr, i) -> arr[i > 0 ? 0 : 1]++, 
  (l, r) -> {  l[0] += r[0]; l[1] += r[1]; });

Answer 5

这里也有一种计算零的方法：

int[] ans = new int[3];
list.forEach(i -> ans[Integer.signum(i) + 1] += 1);

ans 数组现在包含 3 个值：索引 0 表示负计数，索引 1 表示零计数，索引 2 表示正计数。

即对于以下输入：[1, 2, 3, 0, -4, -5]，输出将是：[2, 1, 3]。

---

一种更实用的方式，不包括零：

Map<Integer, Long> bySignum = list.stream()
    .filter(i -> i != 0)
    .collect(Collectors.groupingBy(Integer::signum, Collectors.counting()));

产生此输出：{-1=2, 1=3}。