删除集合中元素选择for循环、Iterator还是foreach?

阿里开发手册也明确说明禁止使用foreach删除、增加List元素。

正确删除集合元素的方式是使用迭代器（Iterator），代码如下：

@Test
void contextLoads() {

    List<String> list = new ArrayList<>(4);
    list.add("a");
    list.add("ab");
    list.add("abc");
    list.add("abcd");

    Iterator<String> iterator = list.iterator();
    while (iterator.hasNext()) {
        if (iterator.next().contains("a")) {
            // 删除元素
            iterator.remove();
        }
    }
    System.out.println(list);//[]
}

JDK8后lambda写法：list.removeIf(s -> s.contains(“a”));

实例

为什么不能使用for循环删除集合中的元素。

需求

一个List集合，元素类型为String，有N个元素，删除这些元素中包含字符’‘a’'的元素。

假设集合内容如下：

List list = new ArrayList<>(4);
list.add(“a”);
list.add(“ab”);
list.add(“abc”);
list.add(“abcd”);

正确答案
先上正确答案

public static void main(String[] args) {
    List<String> list = new ArrayList<>(4);
    list.add("a");
    list.add("ab");
    list.add("abc");
    list.add("abcd");
    
    Iterator<String> iterator = list.iterator();
    while (iterator.hasNext()) {
        if (iterator.next().contains("a")) {
            // 删除元素
            iterator.remove();
        }
    }
    System.out.println(list);
}

输出结果为

[]

错误答案1：普通for循环（for-i）

public static void main(String[] args) {
    List<String> list = new ArrayList<>(4);
    list.add("a");
    list.add("ab");
    list.add("abc");
    list.add("abcd");

    for (int i = 0; i < list.size(); i++) {
        if (list.get(i).contains("a")) {
            list.remove(i);
        }
    }
    System.out.println(list);
}

输出结果为

[ab, abcd]

分析

普通for循环遍历List集合的同时，删除List中的元素是可以运行的代码，但在大多数场景下，不能使用这种方式，上边的结果也印证了这一点，虽然代码不会报错，运行也正常，但在本实例中，这么写就是BUG。

BUG原因：索引为i的元素删除后，后边元素的索引自动向前补位，即原来索引为i+1的元素，变为了索引为i的元素，但是下一次循环取的索引是i+1，此时你以为取到的是原来索引为i+1的元素，其实取到是原来索引为i+2的元素。

发现，只要每删除一个元素，就会漏掉下一个元素，所以这种方式从逻辑上来说是存在bug的，无论什么需求场景，都不建议用这种方式。

for 循环正序遍历删除元素是程序BUG,逆序删除元素会有什么结果呢，请看下面代码演示：

@Test
void contextLoads02() {

    List<String> list = new ArrayList<>(4);
    list.add("a");
    list.add("ab");
    list.add("abc");
    list.add("abcd");

    for (int i = list.size()-1; i >=0; i--) {
        if (list.get(i).contains("a")) {
            list.remove(i);
        }
    }
    System.out.println(list);//[]
}

运行结果： []

结论： for 循环逆序遍历删除元素是可以的。

错误答案2：增强for循环（foreach）

   @Test
    void contextLoads03() {
        List<String> list = new ArrayList<>(4);
        list.add("a");
        list.add("ab");
        list.add("abc");
        list.add("abcd");
        for (String str :list) {
            if (str.contains("a")) {
                list.remove(str);
            }
        }
        System.out.println(list);
    }

运行结果：

java.util.ConcurrentModificationException
at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:901)atjava.util.ArrayList$Itr.next(ArrayList.java:851)
at com.cloud.RedisLockApplicationTests.contextLoads03(JspDemoApplicationTests.java:111)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)

Concurrent Modification Exception：并发修改异常