【搬砖系列】如何在遍历List时安全删除集合元素

 

public void testIterRemove() {
        List l1 = new ArrayList(Arrays.asList(1, 2, 3, 4, 5));
        Iterator<Integer> iterator = l1.iterator();
        System.out.println("before=" + l1);
        while (iterator.hasNext()) {
            iterator.next();
            iterator.remove();
        }
        System.out.println("after=" + l1);
}
/*
before=[1, 2, 3, 4, 5]
after=[]
*/

代码中关键的部分在于：

iterator.next();

和

iterator.remove();

二者缺一不可。当然调用next()之前要调用hasNext()确保没有越界。

简单分析两个方法的代码即可得出答案：

首先看remove()方法，为什么hasNext() + remove()不可以呢：

        public void remove() {
            if (lastRet < 0)
                throw new IllegalStateException();
            checkForComodification();

            try {
                ArrayList.this.remove(lastRet);
                cursor = lastRet;
                lastRet = -1;
                expectedModCount = modCount;
            } catch (IndexOutOfBoundsException ex) {
                throw new ConcurrentModificationException();
            }
        }

由于代码用属性lastRet（即lastReturn index）控制异常，所以查看ArrayList#Itr的实例初始化语句，lastRet初值是-1，如果单独调用remove()会报错。

再看一下next()方法，就会找出答案：

        public E next() {
            checkForComodification();
            int i = cursor;
            if (i >= size)
                throw new NoSuchElementException();
            Object[] elementData = ArrayList.this.elementData;
            if (i >= elementData.length)
                throw new ConcurrentModificationException();
            cursor = i + 1;
            return (E) elementData[lastRet = i];
        }

在方法中首先将cursor记录到变量 i，返回前将 i 的知赋给 lastRet ，完成lastRet的更新。方法返回值不是本次要考虑的范畴。

再回过头看remove()方法，有了lastRet后，就可以重置cursor。在方法中调用了ArrayList#remove()，这个方法会重新调整数组元素的位置，如果删掉的不是最后一个位置的元素（numMoved == 0 表示删掉的正好是最后一位元素），将会把index后面的元素往前移动一个，最后将elementData[--size] = null 为移除旧元素,帮助GC。

 

总结：

需要先调用next()更新lastRet的值，才能在remove()中通过第一关对lastRet的检查。因此在遍历List时需要联合使用 ：

while( it.hasNext() ) 
{ 
    it.next(); 
    it.remove();
} 

 

同理，其他集合删除应该也是这个步骤，没来得及验证，欢迎评论区小锤换大锤:)。