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

Question

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

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

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

我猜这是因为每次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);

`partitioningBy` 方法值得我 +1。作为补充，如果 `arrA` 是可变的（正如最初的 `remove` 方法所建议的那样），`arrA = arrA.stream().filter(x ->whatever).toList()` 可以替换为 `arrA.removeIf (x -> !whatever);`... (2认同)

Answer 2

正如其他人所提到的，这对于foreach- 来说是不可能的，因为循环不可能for (A a: arrA)删除元素。

在我看来，最干净的解决方案是使用while带有迭代器的普通 for - 迭代器允许您在迭代时删除元素（只要集合支持）。

Iterator<A> it = arrA.iterator()
while (it.hasNext()) {
    A a = it.next();
    if (!check(a))
        continue;
    arrB.add(a);
    it.remove();
}

这也使您免于复制/克隆arrA。