美文网首页
【java基础-实战3】list遍历时删除元素的方法

【java基础-实战3】list遍历时删除元素的方法

作者: 程序员小2 | 来源:发表于2023-10-21 09:38 被阅读0次

    在实际的业务开发中,容器的遍历可以说是非常非常常见的场景了,遍历删除呢,用的机会也比较多,那么有哪几种删除元素的方法呢?你用对了吗~
    本文循序渐进,先说几种容易出问题的方法,再引出几种比较可靠的方法~

    首先,初始化一个数组,用于后面的事例演示:

    List<Integer> list = new ArrayList<>(); 
    for (int i = 1; i < 5; i++) {
        if(i==2) {
          //i==2时添加两次,用于后面的测试
          list.add(i); 
          list.add(i); 
        }else {
          list.add(i); 
        }
    }
    

    方法一:for-each循环删除(结果:抛出异常)

    for (String id : list){
       if (id.contains(3)) {
          list.remove(id); 
        }
     }
    

    运行上面的代码,抛出如下异常:


    image.png

    抛出异常的根本原因在于for-each是使用Iterator来实现遍历的,调用ArrayList.remove()方法会将modCount+1,而Iterator内部的expectedModCount确没有更新,这样在进行下次循环时调用Iterator.next()会对modCount和expectedModCount进行比较,不一致就会抛出ConcurrentModificationException异常。
    当删除完元素后,进行下一次循环时,会调用下面源码中Itr.next()方法获取下一个元素,会调用checkForComodification()方法对ArrayList进行校验,判断在遍历ArrayList是否已经被修改,由于之前对modCount+1,而Iterator中的expectedModCount还是初始化时ArrayList.Itr对象时赋的值,所以会不相等,然后抛出ConcurrentModificationException异常。

    方法二:普通for循环正序删除(结果:会漏掉对后一个元素的判断)

    for (int i = 0; i < list.size(); i++) {
        if (2==equals(list.get(i) )) {//2是要删除的元素
            list.remove(i);
            //解决方案: 加一行代码i = i - 1; 删除元素后,下标减1
        }
        System.out.println("当前List是"+list.toString());
    }
    //原ArrayList是[1, 2, 3, 3, 4]
    //删除后是[1, 2, 3, 4], 少删除了一个元素2
    
    

    可以看到少删除了一个元素"2".

    原因在于调用remove删除元素时,remove方法调用System.arraycopy()方法将后面的元素移动到前面的位置,也就是第二个num:2会移动到数组下标为2的位置,而在下一次循环时,i+1之后,i会为2,不会对数组下标为1这个位置进行判断,所以这种写法,在删除元素时,被删除元素a的后一个元素b会移动a的位置,而i已经加1,会忽略对元素b的判断,所以如果是连续的重复元素,会导致少删除。
    解决方案:可以在删除元素后,执行i=i-1,使得下次循环时再次对该数组下标进行判断。

    方法三:普通for循环倒序删除(结果:正确删除)

     for (int i = list.size() -1 ; i>=0; i--) {
        if (list.get(i).equals(2)) {
            list.remove(i);
        }
        System.out.println("当前list是"+list.toString());
    }
    
    
    //原ArrayList是[1, 2, 3, 3, 4]
    //删除后是[1, 3, 4]
    

    这种方法可以正确删除元素,因为调用remove删除元素时,remove方法调用System.arraycopy()将被删除元素a后面的元素向前移动,而不会影响元素a之前的元素,所以倒序遍历可以正常删除元素。

    方法四:Iterator遍历,使用ArrayList.remove()删除元素(结果:抛出异常)

    Iterator<Integer> iterator = list.iterator();
    while (iterator.hasNext()) {
        Integer value = iterator.next();
        if (value.equals(3)) {//3是要删除的元素
                list.remove(value);
        }
        System.out.println("当前list是"+list.toString());
    }
    
    

    第4种方法其实是第1种方法在编译后的代码,所以第四种写法也会抛出ConcurrentModificationException异常。这种需要注意的是,每次调用iterator的next()方法,会导致游标向右移动,从而达到遍历的目的。所以在单次循环中不能多次调用next()方法,不然会导致每次循环时跳过一些元素.

    方法五: Iterator遍历,使用Iterator的remove删除元素(结果:正确删除)

    Iterator<Integer> iterator = list.iterator();
    while (iterator.hasNext()) {
        Integer value = iterator.next();
        if (value.equals(3)) {//3是需要删除的元素
            iterator.remove();
        }
    }
    

    方法5可以正确删除元素。

    跟第1种和第4种方法的区别在于是使用iterator.remove();来移除元素,而在remove()方法中会对iterator的expectedModCount变量进行更新,所以在下次循环调用iterator.next()方法时,expectedModCount与modCount相等,不会抛出异常。

    方法六:jdk8+ 流方式 list.removeIf (结果:正确删除)

    jdk8+ 推荐下面这种写法,简洁明了

    list.removeIf(s -> s.contains(3));
    

    结论:

    在list遍历中不要使用list.remove(), 容易出问题;
    推荐使用方法五的iterator.remove()或者方法六的 list.removeIf().

    相关文章

      网友评论

          本文标题:【java基础-实战3】list遍历时删除元素的方法

          本文链接:https://www.haomeiwen.com/subject/hmdkidtx.html