使用Stream API根据条件将列表拆分为子列表

Question

我有一个具体的问题。有一些类似的问题，但是要么是Python的问题，要么是Java的问题，或者即使问题听起来很相似，要求也有所不同。

我有一个值列表。

List1 = {10, -2, 23, 5, -11, 287, 5, -99}

归根结底，我想根据列表的值拆分列表。我的意思是，如果该值大于零，它将保留在原始列表中，并且负值列表中的相应索引将被设置为零。如果该值小于零，它将转到负值列表，原始列表中的负值将替换为零。

结果列表应该像这样；

List1 = {10, 0, 23, 5, 0, 287, 5, 0}
List2 = {0, -2, 0, 0, -11, 0, 0, -99}

有没有办法用Java中的Stream api解决此问题？

Answer 1

如果要在单个Stream操作中执行此操作，则需要一个自定义收集器：

List<Integer> list = Arrays.asList(10, -2, 23, 5, -11, 287, 5, -99);

List<List<Integer>> result = list.stream().collect(
    () -> Arrays.asList(new ArrayList<>(), new ArrayList<>()),
    (l,i) -> { l.get(0).add(Math.max(0, i)); l.get(1).add(Math.min(0, i)); },
    (a,b) -> { a.get(0).addAll(b.get(0)); a.get(1).addAll(b.get(1)); });

System.out.println(result.get(0));
System.out.println(result.get(1));