当要删除ArrayList里面的某个元素,一不注意就容易出bug。今天就给大家说一下在ArrayList循环遍历并删除元素的问题。首先请看下面的例子:
importjava.util.ArrayList;public classArrayListRemove
{public static voidmain(String[] args)
{
ArrayList list = new ArrayList();
list.add("a");
list.add("b");
list.add("b");
list.add("c");
list.add("c");
remove(list);for(String s : list)
{
System.out.println("element : " +s);
}
}public static void remove(ArrayListlist)
{
}
}
常见错误写法:
一:
public static void remove(ArrayListlist)
{for (int i = 0; i < list.size(); i++)
{
String s=list.get(i);if (s.equals("b"))
{
list.remove(s);
}
}
}
结果:第二个“b”的字符串没有删掉。
二:
public static void remove(ArrayListlist)
{for(String s : list)
{if (s.equals("b"))
{
list.remove(s);
}
}
}
结果:这种for-each写法会报出并发修改异常:java.util.ConcurrentModificationException。
错误原因:先看下ArrayList中的remove方法,看入参为Object的remove方法是怎么实现的:
public booleanremove(Object o) {if (o == null) {for (int index = 0; index < size; index++)if (elementData[index] == null) {
fastRemove(index);return true;
}
}else{for (int index = 0; index < size; index++)if(o.equals(elementData[index])) {
fastRemove(index);return true;
}
}return false;
}
执行路径会到else路径下最终调用faseRemove方法:
private void fastRemove(intindex) {
modCount++;int numMoved = size - index - 1;if (numMoved > 0)
System.arraycopy(elementData, index+1, elementData, index,numMoved);
elementData[--size] = null; //Let gc do its work
}
可以看到会执行System.arraycopy方法,导致删除元素时涉及到数组元素的移动。针对错误写法一,在遍历第一个字符串b时因为符合删除条件,所以将该元素从数组中删除,并且将后一个元素移动(也就是第二个字符串b)至当前位置,导致下一次循环遍历时后一个字符串b并没有遍历到,所以无法删除。针对这种情况可以倒序删除的方式来避免:
public static void remove(ArrayListlist)
{for (int i = list.size() - 1; i >= 0; i--)
{
String s=list.get(i);if (s.equals("b"))
{
list.remove(s);
}
}
}
因为数组倒序遍历时即使发生元素删除也不影响后序元素遍历。
实例二的错误原因。产生的原因却是foreach写法是对实际的Iterable、hasNext、next方法的简写,问题同样处在上文的fastRemove方法中,可以看到第一行把modCount变量的值加一,但在ArrayList返回的迭代器:
public Iteratoriterator() {return newItr();
}
这里返回的是AbstractList类内部的迭代器实现private class Itr implements Iterator,看这个类的next方法:
publicE next() {
checkForComodification();try{
E next=get(cursor);
lastRet= cursor++;returnnext;
}catch(IndexOutOfBoundsException e) {
checkForComodification();throw newNoSuchElementException();
}
}
第一行checkForComodification方法:
final voidcheckForComodification() {if (modCount !=expectedModCount)throw newConcurrentModificationException();
}
这里会做迭代器内部修改次数检查,因为上面的remove(Object)方法修改了modCount的值,所以才会报出并发修改异常。要避免这种情况的出现则在使用迭代器迭代时(显示或for-each的隐式)不要使用ArrayList的remove,改为用Iterator的remove即可。
public static void remove(ArrayListlist)
{
Iterator it =list.iterator();while(it.hasNext())
{
String s=it.next();if (s.equals("b"))
{
it.remove();
}
}
}