文章目录
- for循环
- foreach循环
- iterator遍历
- 总结
面试问到这个,印象中可以用迭代器解决,但是具体原理记不清楚了,整理一下
循环遍历list的三种方式:for循环、foreach循环、iterator遍历。
for循环
代码示例:
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");
for(int i=0;i<list.size();i++){
if(Integer.parseInt(list.get(i)) >=1){
list.remove(i);
}
}
这种方式有问题,问题在于删除某个元素后,list的大小发生变化,同时索引也会发生变化,所以会导致在遍历的时候漏掉某些元素。比如当你删除第1个元素后,继续根据索引访问第2个元素时,因为删除元素后,后面的元素都往前移动了一位,所以实际访问的是第3个元素(debug试一下就知道了)。因此,这种方式可以用在删除特定的一个元素时使用,但不适合循环删除多个元素时使用。
解决方法:
- 每遍历完一个索引,回退一步,这样可以避免跳过进位的数据
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");
for(int i=0;i<list.size();i++){
if(Integer.parseInt(list.get(i)) >=1){
list.remove(i);
i--;
}
}
- 从后面往前遍历
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");
for(int i=list.size();i>=0;i--){
if(Integer.parseInt(list.get(i)) >=1){
list.remove(i);
}
}
foreach循环
代码示例:
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("3");
for(String str:list){
if(Integer.parseInt(str) >=1){
list.remove(str);
}
}
抛出异常:java.util.ConcurrentModificationException
foreach 写法实际上是对的 iterator、hasNext、next方法的简写。因此从List.iterator()源码着手分析,跟踪iterator()方法,该方法返回了 Itr 迭代器对象。
public Iterator<E> iterator() {
return new Itr();
}
Itr类的定义:
private class Itr implements Iterator<E> {
int cursor; // index of next element to return
int lastRet = -1; // index of last element returned; -1 if no such
int expectedModCount = modCount;
Itr() {}
public boolean hasNext() {
return cursor != size;
}
@SuppressWarnings("unchecked")
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];
}
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();
}
}
@Override
@SuppressWarnings("unchecked")
public void forEachRemaining(Consumer<? super E> consumer) {
Objects.requireNonNull(consumer);
final int size = ArrayList.this.size;
int i = cursor;
if (i >= size) {
return;
}
final Object[] elementData = ArrayList.this.elementData;
if (i >= elementData.length) {
throw new ConcurrentModificationException();
}
while (i != size && modCount == expectedModCount) {
consumer.accept((E) elementData[i++]);
}
// update once at end of iteration to reduce heap write traffic
cursor = i;
lastRet = i - 1;
checkForComodification();
}
final void checkForComodification() {
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
}
}
通过代码我们发现 Itr 是 ArrayList 中定义的一个私有内部类,在 next、remove方法中都会调用checkForComodification
方法,该方法的作用是判断 modCount 和 expectedModCount是否相等,如果不相等则抛出ConcurrentModificationException异常。每次正常执行 remove 方法后,都会执行expectedModCount = modCount
,保证两个值相等,那么问题基本上已经清晰了,在 foreach 循环中执行 list.remove(str),对 list 对象的 modCount 值进行了修改,而 list 对象的迭代器的 expectedModCount 值未进行修改,因此抛出了ConcurrentModificationException异常。
iterator遍历
代码示例:
Iterator<String> it = list.iterator();
while(it.hasNext()){
String x = it.next();
if(Integer.parseInt(x) >=1){
it.remove();
}
}
这种方式可以正常的循环及删除。但要注意的是,使用iterator的remove方法,不能使用list的remove方法。
总结
- 用for循环遍历List删除元素时,需要注意索引会左移的问题。
- List删除元素时,最好使用迭代器iterator的remove方式。