使用Iterator删除元素

BBPH • 2024-10-26 11:54 • 编程

一、Iterator的使用方法

在Java编程中，使用Iterator进行迭代操作是非常常见的。Iterator提供了一种顺序访问集合中元素的方法，而且不会暴露于底层集合的实现细节。我们可以通过以下步骤来获取Iterator对象：

Iterator<E> iterator = collection.iterator();

其中，collection表示要迭代的集合，E是集合中元素的类型。

获取Iterator对象后，我们可以使用hasNext()方法来判断集合中是否还有元素可以迭代，使用next()方法来获取下一个元素。

while (iterator.hasNext()) {
    E element = iterator.next();
    // 对element进行操作
}

二、删除集合中元素的方法

在Java编程中，我们经常需要从一个集合中删除其中的元素。但是，如果直接在迭代过程中使用集合自带的remove()方法来删除元素，会导致ConcurrentModificationException异常，因为在迭代器遍历集合时，集合结构已经发生了变化。

为了避免这种异常，我们可以使用Iterator的remove()方法来删除元素。该方法会在删除元素的同时更新迭代器的状态，确保迭代器和集合的状态保持一致。具体使用方法如下：

Iterator<E> iterator = collection.iterator();
while (iterator.hasNext()) {
    E element = iterator.next();
    if (满足条件) {
        iterator.remove();
    }
}

当调用iterator.remove()方法时，它会删除集合中迭代器最后返回的元素。

三、使用Iterator删除元素的示例

假设我们有一个List，其中保存了一些学生的年龄，我们要删除其中所有年龄小于18岁的学生。

List<Integer> ages = new ArrayList<>();
ages.add(20);
ages.add(15);
ages.add(25);
ages.add(17);
ages.add(19);

Iterator<Integer> iterator = ages.iterator();
while (iterator.hasNext()) {
    Integer age = iterator.next();
    if (age < 18) {
        iterator.remove();
    }
}
System.out.println(ages); // [20, 25, 19]

运行结果为[20, 25, 19]，删除操作成功。

四、使用forEach循环删除元素的注意事项

Java 8引入了forEach循环，简化了集合的迭代操作。但是，使用forEach循环删除元素要小心，否则会抛出ConcurrentModificationException异常。

在使用forEach循环遍历集合时，不能在循环过程中同时修改集合，只能使用Iterator的remove()方法删除元素。否则，会导致集合结构发生变化，引发ConcurrentModificationException异常。示例代码如下：

List<Integer> ages = new ArrayList<>();
ages.add(20);
ages.add(15);
ages.add(25);
ages.add(17);
ages.add(19);

ages.forEach(age -> {
    if (age < 18) {
        ages.remove(age); // 抛出ConcurrentModificationException异常
    }
});
System.out.println(ages);

运行结果为ConcurrentModificationException异常。

如果需要使用forEach循环删除元素，可以先将要删除的元素保存到一个新的列表中，然后在循环结束后再调用removeAll()方法批量删除元素。示例代码如下：

List<Integer> ages = new ArrayList<>();
ages.add(20);
ages.add(15);
ages.add(25);
ages.add(17);
ages.add(19);

List<Integer> toRemove = new ArrayList<>();
ages.forEach(age -> {
    if (age < 18) {
        toRemove.add(age);
    }
});
ages.removeAll(toRemove);
System.out.println(ages); // [20, 25, 19]

五、总结

Iterator提供了一种顺序访问集合中元素的方法，而且不会暴露于底层集合的实现细节。如果需要在迭代过程中删除元素，可以使用Iterator的remove()方法来删除元素。在使用forEach循环遍历集合时，不能在循环过程中同时修改集合，只能使用Iterator的remove()方法删除元素。

原创文章，作者：BBPH，如若转载，请注明出处：https://www.506064.com/n/144896.html