循环删除 List 中的元素,这个问题是有需要的注意点的,如果是个新手,确实会遇到一点麻烦
新手常犯的错误
可能很多新手(包括现在的我,哈哈)第一时间想到的写法是下面这样的:
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 异常!
使用 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 次循环将元素"掘金"删除后,元素的下标变成了下面这样:
stream filter 过滤
public static void main(String[] args) {
List<String> platformList = new ArrayList();
platformList.add("博客园");
platformList.add("CSDN");
platformList.add("掘金");
platformList = platformList.stream().filter(e -> !e.startsWith("博客园")).collect(Collectors.toList());
System.out.println(platformList);
}
输出结果:
这个方法即是利用了 Stream 的筛选功能,快速过滤所需要的元素,虽然不是进行集合删除,但达到了同样的目的,这种方法要更简洁吧。