使用Java流删除和收集元素

Question

假设我有一个集合，还有一个谓词，它匹配我想从集合中删除的元素。但我不只是想丢弃它们，我想将匹配的元素移动到一个新集合中。在Java 7中，我会这样做：

List<E> removed = new LinkedList<>();
for (Iterator<E> i = data.iterator(); i.hasNext();) {
    E e = i.next();
    if (predicate.test(e)) {
        removed.add(e);
        i.remove();
    }
}

我很好奇是否有一个流/Java8方法来做这件事。Collections.remove如果（）不幸地只是返回一个布尔值（甚至没有删除元素的数量？太糟糕了。）我设想了这样的东西（当然. remveAndYield（谓语）不存在）：

List<E> removed = data.removeAndYield(predicate).collect(Collectors.toList());

注：此问题受类似问题启发；这个问题是关于在从集合中移除的项上获取流的更一般的情况。正如链接问题中所指出的，命令式解决方案可能更具可读性，但我很好奇，这在流中是否可行。

编辑：显然，我们可以将任务分成两个独立的步骤，假设适当的数据结构将是有效的。问题是可以在任意集合（可能没有有效的。包含（）等）上执行此操作吗？

Answer 1

如果您想要一种函数式的方法来实现这一点，您可以编写自己的方法。

static <E> Set<E> removeIf(Collection<? extends E> collection, Predicate<? super E> predicate) {
    Set<E> removed = new HashSet<>();
    for (Iterator<? extends E> i = collection.iterator(); i.hasNext();) {
        E e = i.next();
        if (predicate.test(e)) {
            removed.add(e);
            i.remove();
        }
    }
    return removed;
}

这可用于从列表中删除所有奇数。

Set<Integer> set = new HashSet<>(Arrays.asList(1, 2, 3, 4, 5, 6));
Set<Integer> removed = removeIf(set, i -> i % 2 != 0);
System.out.println(set);
System.out.println(removed);

Answer 2

我会保持简单：

Set<E> removed = set.stream()
    .filter(predicate)
    .collect(Collectors.toSet());

set.removeAll(removed);

Answer 3

如果您不介意，让我稍微改变一下您的要求。：-）

所需结果的一个特征是匹配的元素应该在一个集合中结束，而非匹配的元素应该在不同的集合中结束。在Java-8之前的变体世界中，考虑获取非匹配元素集合的最简单方法是从原始集合中移除匹配元素。

但是，删除——对原始列表的修改——是需求的固有部分吗？

如果不是，那么可以通过一个简单的分区操作来实现结果：

Map<Boolean, List<E>> map = data.stream().collect(partitioningBy(predicate));

结果映射本质上是两个列表，其中包含匹配（key=true）和非匹配（key=false）元素。

优点是这种技术可以一次完成，必要时可以并行完成。当然，与从原始元素中删除匹配项相比，这会创建一个非匹配元素的重复列表，但这是为不变性付出的代价。这种权衡可能是值得的。