后端开发小白

Java ArrayList在for循环内删除元素的问题

2020-01-07  本文已影响0人  Nikfce

Java对ArrayList的遍历方式有很多种,for-index, for-each, iterable.forEach, Iterator等,这里着重要谈谈for-each方式和在循环中删除时会发生什么。

首先先谈for-index为什么不能在循环中删除。(先上代码

// list = {1, 2, 3, 4, 5}

for (int i = 0 ; i < list.size() ; i ++) {

    if (2 == l.get(i)) {

        list.remove(i);

    }

}

通过观察源码可以发现,每一次remove()它都会调用System.copyarray()拷贝一次,也就是说把'2'给删除,且'2'之后的元素都往前移动一位,因此'2'之后的'3'是遍历不到的,这样直接remove()很容易出问题。

遍历中删除正确的姿势是利用Iterator,(上代码

Iterator<Integer> iterator = list.iterator();

while (iterator.hasNext())

{

    int tmp = iterator.next();

    if (tmp == 2) {

        iterator.remove();

    }

}

当然Collection接口提供了removeIf的方法可以代替这么长的代码,这里贴出来只是为了方便解释之后的for-each方法。

我们知道for-each方法循环其实也是通过Iterator实现的,但是它是不支持循环内删除的,如果调用list.remove()就会触发ConcurrentModificationException。(还是先贴代码吧

// list = {1, 2, 3, 4, 5}

for (int a : list) {

    if (a == 1) {

        list.remove((Integer) a); // 这里因为要调用得失remove(Object o)方法而不是remove(int index)方法,其实调用哪个方法都一样

    }

}

循环开始就会触发ConcurrentModificationException,这是因为调用remove()方法会将modCount++,而iterator调用next()获得下一个元素时会检查modCount是否和创建Iterator时(for循环开始时)一样,如果不一样就会抛ConcurrentModificationException错误。这是ArrayList内部元素的一个保护机制。

知道了这些以后,下面就要说点有意思的了,如果对上面的代码进行一点修改,可能结果时你没想到的:

// list = {1, 2, 3, 4, 5}

for (int a : list) {   

    System.out.println(a);

    if (a == 4) {       

        list.remove((Integer) a);

    }

}

程序居然没有报错😐,而且程序的输出是1,2,3,4, 也就是说5没有被遍历到。有兴趣的同学可能会去尝试把 a==4 改成 a==其他数 但是发现只有a==4没报错,这是因为for-each循环的iterator在调用next()之前会调用hasNext(),而hasNext()的实现是

return cursor != size;

当调用list.remove()之后,数组的size会减1,而由于4是数组的倒数第二个元素,cursor原本等于size - 1的,但是因为调用了list.remove()导致cursor == size, 使得hasNext()返回了false,因此没有进入到next()方法,不会打印出5, 也就不会有ConcurrentModificationException报错啦。

上一篇下一篇

猜你喜欢

热点阅读