多个流过滤器或 forEach if-else

Question

我有一个小列表，我必须根据元素的值将其分成 2 个列表。我正在寻找两种方法来做到这一点 -

1. 使用过滤器对列表进行两次迭代。

List<String> sublist1 = list.stream().filter(condition1).collect(ImmutableList.toImmutableList());
List<String> sublist2 = list.stream().filter(condition2).collect(ImmutableList.toImmutableList());

1. 在流​​中使用 forEach 来分配值

List<String> sublist1 = new LinkedList<>();
List<String> sublist2 = new LinkedList<>();
list.stream().forEach(element -> {
    if(condition1) sublist1.add(element);
    else if (condition2) sublist2.add(element);
})

我想知道哪种方式更好更有效地实现这一点？

Answer 1

如果您有两个明确的条件导致两个子列表，并且它们是替代，这两个条件中只有一个可能为真，您可以在filter 或Predicate.or 中使用 OR然后在任一条件下使用Collectors.partitioningBy：

Map<Boolean, List<String>> sublistMap = list
        .stream()
        .filter(condition1.or(condition2))
        .collect(Collectors.partitioningBy(condition1, Collectors.toUnmodifiableList()));

List<String> sublist1 = sublistMap.get(Boolean.TRUE); // condition1
List<String> sublist2 = sublistMap.get(Boolean.FALSE); // condition2

测试

List<String> list = Arrays.asList("a", "b", "cc", "ddd", "aaaa", "vvv", "oo");
Predicate<String> condition1 = (s) -> s.contains("o");
Predicate<String> condition2 = (s) -> s.contains("a");

System.out.println(sublist1);
System.out.println(sublist2);

输出：

[oo]
[a, aaaa]