如何在遍历时删除List中的元素

66 阅读 0 评论 44 点赞

我是靠谱客的博主俏皮镜子，最近开发中收集的这篇文章主要介绍如何在遍历时删除List中的元素，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

转自：http://uin57.iteye.com/blog/903966

今天在写代码的时候突然脑袋短路了，不知怎么在遍历的时候删除List中的元素了。

于是就产生了下面一堆代码。。

代码1：foreach

package testCollection.testList;
import java.util.ArrayList;
public class TestRemoveList {
public static void main(String[] args) {
ArrayList<TestObject> testList = new ArrayList();
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(null);
for(TestObject to:testList){
if(null==to){
testList.remove(to); //这一行改变了集合内容
}
}
}
}
class TestObject{}

运行结果：

Exception in thread "main" java.util.ConcurrentModificationException
at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:819)
at java.util.ArrayList$Itr.next(ArrayList.java:791)
at testCollection.testList.TestRemoveList.main(TestRemoveList.java:13)

分析一下原因：

在List类中有如下变量：

protected transient int modCount = 0;

这个变量是用来记录这个List是否改变(这里只说ArrayList)

比如add方法：


public boolean add(E e) {
ensureCapacityInternal(size + 1);
// Increments modCount!!
elementData[size++] = e;
return true;
}

继续进入查看ensureCapacityInternal方法：


private void ensureCapacityInternal(int minCapacity) {
modCount++;
// overflow-conscious code
if (minCapacity - elementData.length > 0)
grow(minCapacity);
}

看到了吧，modCount++了

比如remove方法，这个方法会调用fastRemove方法，这个方法才是真正改变list里面的内容：


private void fastRemove(int index) {
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
}

看到了吧 modCount++了

foreach方法调用的是iterator()中返回的Iterator实现类：


public Iterator<E> iterator() {
return new Itr();
}

就在你调用这个方法的时候，会创建一个Itr对象

在创建的时候就会初始化expectedModCount变量，这个变量的值等于List外部类中的modCount

int expectedModCount = modCount;

如果你不是使用Iterator方法该边List的话，List中的modCount值是变了，但是这里的expectedModCount可没有变化

所以当调用Itr对象的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];
}


final void checkForComodification() {
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
}

就会发生悲剧。

于是乎：java.util.ConcurrentModificationException。

JDK6手册上原话是这样子的：

JDK6 API 写道

public class ConcurrentModificationExceptionextends RuntimeException当方法检测到对象的并发修改，但不允许这种修改时，抛出此异常。
例如，某个线程在 Collection 上进行迭代时，通常不允许另一个线性修改该 Collection。通常在这些情况下，迭代的结果是不确定的。如果检测到这种行为，一些迭代器实现（包括 JRE 提供的所有通用 collection 实现）可能选择抛出此异常。执行该操作的迭代器称为快速失败迭代器，因为迭代器很快就完全失败，而不会冒着在将来某个时间任意发生不确定行为的风险。
注意，此异常不会始终指出对象已经由不同线程并发修改。如果单线程发出违反对象协定的方法调用序列，则该对象可能抛出此异常。例如，如果线程使用快速失败迭代器在 collection 上迭代时直接修改该 collection，则迭代器将抛出此异常。
注意，迭代器的快速失败行为无法得到保证，因为一般来说，不可能对是否出现不同步并发修改做出任何硬性保证。快速失败操作会尽最大努力抛出 ConcurrentModificationException。因此，为提高此类操作的正确性而编写一个依赖于此异常的程序是错误的做法，正确做法是：ConcurrentModificationException 应该仅用于检测 bug。

Java中的For each实际上使用的是iterator进行处理的。而iterator是不允许集合在iterator使用期间删除的。而我在for each时，从集合中删除了一个元素，这导致了iterator抛出了ConcurrentModificationException。

代码2：get（i）

package testCollection.testList;
import java.util.ArrayList;
public class TestRemoveList2 {
public static void main(String[] args) {
ArrayList<TestObject> testList = new ArrayList();
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(null);
testList.add(null);
for (int i = 0; i < testList.size(); i++) {
TestObject to = testList.get(i);
if (null == to) {
System.out.println("移除之前：" + testList.size());
testList.remove(i); // 这一行改变了集合内容
/*
* 当遍历到第五个元素的时候，i=4,testList的长度可悲的变成5(本来是6的)
* 执行完这个循环之后，i++了，变成5了，5不小于5所以不循环，最后一个null元素没有移除
*/
System.out.println("当移除一个元素的时候：" +testList.size() +" "+ i);
}
}
for(TestObject to : testList){
System.out.println(to);
}
}
}

运行结果：

移除之前：6
当移除一个元素的时候：5 4
testCollection.testList.TestObject@a0002df
testCollection.testList.TestObject@6bc839a
testCollection.testList.TestObject@4263f6ea
testCollection.testList.TestObject@30e79eb3
null

换种做法：

package testCollection.testList;
import java.util.ArrayList;
public class TestRemoveList3 {
public static void main(String[] args) {
ArrayList<TestObject> testList = new ArrayList();
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(null);
testList.add(null);
int length = testList.size();
for (int i = 0; i < length; i++) {//这样会越界
TestObject to = testList.get(i);
if (null == to) {
testList.remove(i);
}
}
for(TestObject to : testList){
System.out.println(to);
}
}
}

运行结果：

Exception in thread "main" java.lang.IndexOutOfBoundsException: Index: 5, Size: 5
at java.util.ArrayList.rangeCheck(ArrayList.java:604)
at java.util.ArrayList.get(ArrayList.java:382)
at testCollection.testList.TestRemoveList3.main(TestRemoveList3.java:16)

于是乎：过滤的数据不完全。当有两个连续的元素为null的时候就会有漏网之鱼。

原因在于：集合删除元素时，集合的size会变小，连带索引都会改变！

代码3：iterator

package testCollection.testList;
import java.util.ArrayList;
import java.util.Iterator;
public class TestRemoveList4 {
public static void main(String[] args) {
ArrayList<TestObject> testList = new ArrayList();
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(new TestObject());
testList.add(null);
testList.add(null);
for (Iterator<TestObject> it = testList.iterator();it.hasNext();){
TestObject to=it.next();
if(to==null ){
it.remove();
}
}
for(TestObject to : testList){
System.out.println(to);
}
}
}

运行结果：

testCollection.testList.TestObject@3c0f3387
testCollection.testList.TestObject@a0002df
testCollection.testList.TestObject@6bc839a
testCollection.testList.TestObject@4263f6ea

这次正确了

JDK手册中关于iterator中remove()方法的描述写道