面试急转弯List如何一边遍历一边删除

42次阅读

共计 2663 个字符，预计需要花费 7 分钟才能阅读完成。

来源：juejin.im/post/5e93ce3be51d45470d528262

这是最近面试时被问到的 1 道面试题，本篇博客对此问题进行总结分享。

新手常犯的错误

可能很多新手（包括当年的我，哈哈）第一时间想到的写法是下面这样的：

public static void main(String[] args) {List<String> platformList = new ArrayList<>();
    platformList.add("博客园");
    platformList.add("CSDN");
    platformList.add("掘金");

    for (String platform : platformList) {if (platform.equals("博客园")) {platformList.remove(platform);
        }
    }

    System.out.println(platformList);
}

然后满怀信心的去运行，结果竟然抛 java.util.ConcurrentModificationException 异常了，翻译成中文就是：并发修改异常。

是不是很懵，心想这是为什么呢？

让我们首先看下上面这段代码生成的字节码，如下所示：

由此可以看出，foreach 循环在实际执行时，其实使用的是 Iterator，使用的核心方法是 hasnext() 和 next()。

然后再来看下 ArrayList 类的 Iterator 是如何实现的呢？

可以看出，调用 next() 方法获取下一个元素时，第一行代码就是调用了 checkForComodification();，而该方法的核心逻辑就是比较 modCount 和 expectedModCount 这 2 个变量的值。

在上面的例子中，刚开始 modCount 和 expectedModCount 的值都为 3，所以第 1 次获取元素 ” 博客园 ” 是没问题的，但是当执行完下面这行代码时：

platformList.remove(platform);
modCount 的值就被修改成了 4。

所以在第 2 次获取元素时，modCount 和 expectedModCount 的值就不相等了，所以抛出了 java.util.ConcurrentModificationException 异常。

既然不能使用 foreach 来实现，那么我们该如何实现呢？

主要有以下 3 种方法：

使用 Iterator 的 remove() 方法

使用 for 循环正序遍历

使用 for 循环倒序遍历

接下来一一讲解。

使用 Iterator 的 remove() 方法

使用 Iterator 的 remove() 方法的实现方式如下所示：

public static void main(String[] args) {List<String> platformList = new ArrayList<>();
    platformList.add("博客园");
    platformList.add("CSDN");
    platformList.add("掘金");

    Iterator<String> iterator = platformList.iterator();
    while (iterator.hasNext()) {String platform = iterator.next();
        if (platform.equals("博客园")) {iterator.remove();
        }
    }

    System.out.println(platformList);
}

输出结果为：

[CSDN, 掘金]
为什么使用 iterator.remove(); 就可以呢？

让我们看下它的源码：

可以看出，每次删除一个元素，都会将 modCount 的值重新赋值给 expectedModCount，这样 2 个变量就相等了，不会触发 java.util.ConcurrentModificationException 异常。更多面试题，欢迎关注公众号 Java 面试题精选

使用 for 循环正序遍历

使用 for 循环正序遍历的实现方式如下所示：

public static void main(String[] args) {List<String> platformList = new ArrayList<>();
    platformList.add("博客园");
    platformList.add("CSDN");
    platformList.add("掘金");

    for (int i = 0; i < platformList.size(); i++) {String item = platformList.get(i);

        if (item.equals("博客园")) {platformList.remove(i);
            i = i - 1;
        }
    }

    System.out.println(platformList);
}

这种实现方式比较好理解，就是通过数组的下标来删除，不过有个注意事项就是删除元素后，要修正下下标的值：

i = i – 1;
为什么要修正下标的值呢？

因为刚开始元素的下标是这样的：

第 1 次循环将元素 ” 博客园 ” 删除后，元素的下标变成了下面这样：

第 2 次循环时 i 的值为 1，也就是取到了元素”掘金“，这样就导致元素 ”CSDN” 被跳过检查了，所以删除完元素后，我们要修正下下标，这也是上面代码中 i = i – 1; 的用途。更多面试问题可以关注微信订阅号码匠笔记回复面试获取

使用 for 循环倒序遍历

使用 for 循环倒序遍历的实现方式如下所示：

public static void main(String[] args) {List<String> platformList = new ArrayList<>();
    platformList.add("博客园");
    platformList.add("CSDN");
    platformList.add("掘金");

    for (int i = platformList.size() - 1; i >= 0; i--) {String item = platformList.get(i);

        if (item.equals("掘金")) {platformList.remove(i);
        }
    }

    System.out.println(platformList);
}

这种实现方式和使用 for 循环正序遍历类似，不过不用再修正下标，因为刚开始元素的下标是这样的：

第 1 次循环将元素 ” 掘金 ” 删除后，元素的下标变成了下面这样：

第 2 次循环时 i 的值为 1，也就是取到了元素”CSDN“，不会导致跳过元素，所以不需要修正下标。