List如何一边遍历，一边删除？

Posted 2022-12-01 石头wang

背景

本文探讨对一个List遍历，并且遍历过程有删除操作，什么才是正确的做法

测试

直接上代码

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class LoopList 
    // 本文探讨对一个List遍历，并且遍历过程有删除操作，什么才是正确的做法
    public static void main(String[] args) 
        // 在 "既有遍历又有删除" 的场景，使用foreach，会发生 ConcurrentModificationException 异常
        //test1();

        // 在 "既有遍历又有删除" 的场景，使用 "for(int i" 这种方式遍历，也会发生意料外的事情
        //test2();

        // 在 "既有遍历又有删除" 的场景，使用 "for(int i"，甚至会发生数组越界
        //test3();

        // 在 "既有遍历又有删除" 的场景，list.forEach 这种写法，也报ConcurrentModificationException 异常
        //test4();

        // 其他的场景不再举例，直接说下结论，在 "既有遍历又有删除" 的场景，最好就是用Iterator的remove()方法，详细参考：https://www.cnblogs.com/zwwhnly/p/12530819.html
        //test5();

        // 其实这么写也是OK的，使用 removeIf，点开源码，可以看到其实底层也是用了Iterator的remove()方法
        test6();
    


    public static void test1() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("two");
        list.add("three");

        System.out.println("list,before:" + list);
        for (String s : list) 
            if ("one".equals(s)) 
                list.remove(s);
            
        
        System.out.println("list,after:" + list);
    

    public static void test2() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("one");
        list.add("two");
        list.add("three");

        // 可以看到最终处理后的结果并不是 two three，因为i下标没有做调整，第二个one被跳过了，这就是坑！！！
        System.out.println("list,before:" + list);
        for (int i = 0; i < list.size(); i++) 
            String element = list.get(i);
            System.out.println("do sth biz:" + element + ",index:" + i);
            if ("one".equals(element)) 
                list.remove(i);
            
        
        System.out.println("list,after:" + list);
    
    public static void test3() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("two");
        list.add("three");

        // 这种情况还会发生数组越界，因为list.size的值被固定下来了
        // 在 "既有遍历又有删除" 的场景，不要用 for i，坑多！！！
        System.out.println("list,before:" + list);
        int size = list.size();
        for (int i = 0; i < size; i++) 
            String element = list.get(i);
            System.out.println(element);
            if ("one".equals(element)) 
                list.remove(i);
            
        
        System.out.println("list,after:" + list);
    
    public static void test4() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("two");
        list.add("three");

        // 会发生异常
        System.out.println("list,before:" + list);
        list.forEach(element -> 
            if ("one".equals(element)) 
                list.remove(element);
            
        );
        System.out.println("list,after:" + list);
    

    public static void test5() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("one");
        list.add("two");
        list.add("three");

        Iterator<String> itr = list.iterator();
        System.out.println("list,before:" + list);
        while (itr.hasNext()) 
            String element = itr.next();
            if ("one".equals(element)) 
                itr.remove();
            
        
        System.out.println("list,after:" + list);
    

    public static void test6() 
        List<String> list = new ArrayList<>();
        list.add("one");
        list.add("one");
        list.add("two");
        list.add("three");

        Iterator<String> itr = list.iterator();
        System.out.println("list,before:" + list);
        list.removeIf(element -> "one".equals(element));
        System.out.println("list,after:" + list);
    

参考

https://www.cnblogs.com/zwwhnly/p/12530819.html