java - 如何在 foreach 方法中从流中删除对象？

Question

我必须数组:arrA 和 arrB。 arrA 和 arrB 是不同类型的对象列表，add 函数将对象 A 转换为对象 B 。我想将每个对象从 arrA 添加到 arrB 并从 arrA 中删除该对象。我试图通过流来做到这一点:

arrA.stream().foreach(c -> {arrB.add(c); arrA.remove(c);});

当我执行此操作时，会发生两件事:

1. 并非所有对象都从 arrA 传递到 arrB。
2. 在几次迭代后抛出空指针异常。

我猜这是因为数组的长度在每次 remove() 调用后减少并且迭代计数器增加(只有奇数索引下的对象被传递给 arrB)

现在我可以通过在一个流调用中复制数组然后在第二个流调用中删除对象来解决这个问题，但这对我来说似乎不正确。

这个问题的正确解决方案是什么？

编辑。附加信息:在实际实现中，如果之前过滤过这个列表

arrA.stream().filter(some condition).foreach(c -> {arrB.add(c); arrA.remove(c);});

它被调用了几次以将满足不同条件的元素添加到不同的列表(arrC、arrD 等)，但每个对象只能在一个列表中

Answer 1

Streams 旨在以更实用的方式使用，最好将您的集合视为不可变的。

非流方式是:

arrB.addAll(arrA);
arrA.clear();

但是您可能正在使用 Streams，因此您可以过滤输入，使其更像是:

arrB.addAll(arrA.stream().filter(x -> whatever).toList())

然后从 arrA 中删除(感谢@Holgar 的评论)。

arrA.removeIf(x -> whatever)

如果你的谓词很昂贵，那么你可以划分:

Map<Boolean, XXX> lists = arrA.stream()
  .collect(Collectors.partitioningBy(x -> whatever));
arrA = lists.get(false);
arrB = lists.get(true);

或列出更改:

List<XXX> toMove = arrA.stream().filter(x->whatever).toList();
arrA.removeAll(toMove);
arrB.addAll(toMove);