迭代时移除List中的元素的正确方式

一、简介

本篇文章总结一下另外一个很常见的错误:迭代的时候对集合对象进行删除操作的正确使用方式。

二、使用详解

如果遇到需要循环删除List中匹配的元素,相信很多小伙伴第一个想到的办法就是使用for循环遍历,然后使用挨个比较,如果相等则删除元素,即如下面的代码所示:

List<String> list = new ArrayList<>();
list.add("a");
list.add("b");
list.add("c");
list.add("d");
for (int i = 0; i < list.size(); i++) {
    list.remove(i);
}
//[b, d]
System.out.println(list);

上面的代码应该是经常见到的,我们运行程序,发现结果输出为:

[b, d]

惊不惊喜意不意外,list居然没有全部删除掉,其实很多小伙伴们都犯过这种错误,为什么会出现这种情况呢?

当一个元素被移除时,该List的大小(size)就会缩减,同时也改变了索引的指向,也就是上面的代码只会循环两次,长度在不断减少,第一次循环0 < 4 ,第二次循环 1 < 3 ,不满足下一次循环条件 2 < 2,故只有两次循环就结束。所以,在迭代的过程中使用索引,将无法从List中正确地删除多个指定的元素。

那既然for循环不行,那我们换成foreach试试,代码如下:

List<String> list = new ArrayList<>();
list.add("a");
list.add("b");
list.add("c");
list.add("d");
for (String str : list) {
    if ("a".equals(str)) {
        list.remove(str);
    }
}
System.out.println(list);

运行代码,很遗憾,报错了,如图:

麻蛋,为什么呢?

其实,在 foreach循环中,编译器使得 remove()方法先于next()方法被调用,这就导致了ConcurrentModificationException 异常,我们debug源码看一下,在下图位置抛出异常:

 

由此我们知道,正常情况下,next()方法必须在remove()方法之前被调用,这样才能正确移除元素,我们优化代码如下:

List<String> list = new ArrayList<>();
list.add("a");
list.add("b");
list.add("c");
list.add("d");

Iterator<String> iterator = list.iterator();
while (iterator.hasNext()) {
    String next = iterator.next();
    if ("a".equals(next)) {
        iterator.remove();
    }
}
//[b, c, d]
System.out.println(list);

所以,这才是在循环中移除List元素的正确姿势。 明天圣诞节了,今晚平安夜,都吃苹果没,祝小伙伴们平安夜快乐呀。

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页
实付 19.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值