今天,交流群里突然有人问到一个有关ArrayList的问题,之前就有遇到这个问题,知道是for循环遍历ArrayList调用remove导致的问题,也知道使用iterator迭代器调用是没有问题的,但是貌似没有认真研究过,出现问题的真正原因,也就是为什么for循环中调用remove会出问题?为什么使用iterator迭代器就不会出现问题呢?
一、为什么ArrayList在for循环中调用remove函数会出现问题?(出现上述问题的原因)
我们先来查看一下Arraylist中remove(int index)方法的源码:
public E remove(int index) {
rangeCheck(index); //index范围校验
modCount++;
E oldValue = elementData(index); //Arraylist底层是数组实现的,等同于elementData[index]
int numMoved = size - index - 1; // "dadfad" size = 11 ,index = 6 , numMoved = 4;
if (numMoved > 0)
System.arraycopy(elementData, index+1, elementData, index,numMoved); //这句代码是出现问题的关键。
elementData[--size] = null; // clear to let GC do its work
return oldValue;
}
如上图所示,当for循环 i = 6时,此时““dadfadf”字符串长度大于5,会执行remove方法删掉。
remove方法源码中的 System.arraycopy(elementData, index+1, elementData, index,numMoved);的意思就是将elementData数组中 index+1 角标往后的数组往前移动一位。如下图所示。
执行完remove之后,i=6,原本指向的dadfadf被删除后,由于arraycopy函数的作用,i=6指向了dddaaa,此时for循环会执行i++,i=7,指针后移。如图:
接着往后遍历直到for循环结束都不会出现字符串长度大于5的字符串了,此时你会发现“dddaaa”漏删了,漏掉的罪魁祸首就是arraycopy函数。
二、为什么Iterator迭代器就没问题?
我们先查看一下,iterator常用方法的源码;
public boolean hasNext() { // 判断是否有下一个元素
return cursor != size();
}
public E next() { // 返回下一个元素
checkForComodification();
try {
int i = cursor; // cursor记录的是下一个元素,所以调用next时将返回的是cursor对应的元素
E next = get(i); // 记录需要返回的元素
lastRet = i; // 记录当前元素
cursor = i + 1; // 记录下一个元素
return next;
} catch (IndexOutOfBoundsException e) {
checkForComodification();
throw new NoSuchElementException();
}
}
public void remove() { // 移除元素
if (lastRet < 0)
throw new IllegalStateException();
checkForComodification(); // 检查是否有更改,remove或者add
try {
AbstractList.this.remove(lastRet); // 删除当前元素 (1)
if (lastRet < cursor) // 删除了之后指标减1
cursor--; //**********关键点在这********** (2)
lastRet = -1;
expectedModCount = modCount; // 保持版本号一致
} catch (IndexOutOfBoundsException e) {
throw new ConcurrentModificationException();
}
}
正常情况下,有next方法可知:lastRet 比cursor小1,cursor指向的是下一个结点。如图所示:
(1)行代码,删除lastRet角标的元素,删除“dadfadf”,删除之后,如图所示:
再执行代码(2),Cursor–;lastRet = -1; 如图所示
while循环继续执行,
ArrayList<String> list = new ArrayList<String>(Arrays.asList("a","b","c","d"));
Iterator<String> iter = list.iterator();
while(iter.hasNext()){
String s = iter.next();
if(s.length()>5){
iter.remove();
}
}
此时会执行next函数,执行之后,如图:
下一次仍会调用remove方法,将lastRet指向的“dddaaa”删除。迭代器之所以不会出现问题,主要是因为Cursor指针回退了一格。(要想for循环调用remove方法删除没问题的话,执行remove方法之后可以让i–一次。(亲测有效))代码如下:
for (int i = 0; i < arrayList.size(); i++) {
if (arrayList.get(i).length()>5)
{
arrayList.remove(i);
i--;
}
}
总结
Arrylist调用remove删除多个元素时,尽量使用迭代器的remove方法删除。
实在想用for循环,可以在remove方法调用后,执行i–;