码迷,mamicode.com
首页 > 编程语言 > 详细

【Java】List遍历时删除元素的正确方式

时间:2018-03-11 14:26:12      阅读:176      评论:0      收藏:0      [点我收藏+]

标签:cte   do it   next   fbo   boolean   后序   错误   str   turn   

当要删除ArrayList里面的某个元素,一不注意就容易出bug。今天就给大家说一下在ArrayList循环遍历并删除元素的问题。首先请看下面的例子:

import java.util.ArrayList;
public class ArrayListRemove 
{
    public static void main(String[] args)
    {
        ArrayList<String> list = new ArrayList<String>();
        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(ArrayList<String> list) 
    {

    }
}

常见错误写法:

一:

public static void remove(ArrayList<String> list) 
    {
        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(ArrayList<String> list) 
    {
        for (String s : list)
        {
            if (s.equals("b")) 
            {
                list.remove(s);
            }
        }
    }

结果:这种for-each写法会报出并发修改异常:java.util.ConcurrentModificationException。

 

错误原因:先看下ArrayList中的remove方法,看入参为Object的remove方法是怎么实现的:

public boolean remove(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(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
    }

可以看到会执行System.arraycopy方法,导致删除元素时涉及到数组元素的移动。针对错误写法一,在遍历第一个字符串b时因为符合删除条件,所以将该元素从数组中删除,并且将后一个元素移动(也就是第二个字符串b)至当前位置,导致下一次循环遍历时后一个字符串b并没有遍历到,所以无法删除。针对这种情况可以倒序删除的方式来避免:

public static void remove(ArrayList<String> list) 
    {
        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 Iterator<E> iterator() {
    return new Itr();
}

这里返回的是AbstractList类内部的迭代器实现private class Itr implements Iterator,看这个类的next方法:

public E next() {
        checkForComodification();
        try {
            E next = get(cursor);
            lastRet = cursor++;
            return next;
        } catch (IndexOutOfBoundsException e) {
            checkForComodification();
            throw new NoSuchElementException();
        }
    }

第一行checkForComodification方法:

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

这里会做迭代器内部修改次数检查,因为上面的remove(Object)方法修改了modCount的值,所以才会报出并发修改异常。要避免这种情况的出现则在使用迭代器迭代时(显示或for-each的隐式)不要使用ArrayList的remove,改为用Iterator的remove即可。

public static void remove(ArrayList<String> list) 
    {
        Iterator<String> it = list.iterator();
        while (it.hasNext()) 
        {
            String s = it.next();
            if (s.equals("b")) 
            {
                it.remove();
            }
        }
    }

 

【Java】List遍历时删除元素的正确方式

标签:cte   do it   next   fbo   boolean   后序   错误   str   turn   

原文地址:https://www.cnblogs.com/weknow619/p/8543656.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!