• ArrayList中的遍历删除


    ArrayList 中的遍历删除

    在代码编写过程中经常会遇到这样的要求:遍历一个线性表,要求只遍历一遍(时间复杂度\(O(n)\)),删除符合指定条件的元素,且要求空间复杂度 \(O(1)\)

    例如我们有下列数据,要求遍历列表并删除所有偶数。

    List<Integer> myList = new ArrayList<>(Arrays.toList(new Integer[]{2, 3, 5, 8, 10, 9}));
    

    代码1:直接遍历列表并删除(错误)

    初学者可能会直观地认为,我直接一个for循环遍历删除不就好了吗?但实际上这种做法是错误的。

    static void remove1(List<Integer> list) {
        for (int i = 0; i < list.size(); i++) {
            if (list.get(i) % 2 == 0) {
                list.remove(i);
            }
        }
    }
    

    使用上面的测试数据,结果为:[3, 5, 7, 10, 9],其中一个偶数 10 并没有被删除。

    事实上,当删除列表中元素时,列表的 size() 会改变!当第一个 2 被删除时,此时 list.size() 已经从 6 变成了 5,而 i 只会一直向前跑,因此当 8 被删除时,list.size() 为 4,此时 i 已经变为 5,不再满足 i < list.size() 的循环条件,就会退出循环,后面的元素也不会再被处理。

    代码2:控制循环变量的遍历(低效)

    第二种做法是,不采用 for 循环,而是采用 while 循环 + 控制循环变量 i 的做法。

    static void remove2(List<Integer> list) {
        int i = 0;
        int j;
        while (i < list.size()) {
            if (list.get(i) % 2 == 0) {
                j = i--;
                // 此处存在 i < 0 的可能,所以要及时恢复 i 的位置
                if (i < 0) {
                    i = 0;
                }
                list.remove(j); // 注意此处删除的是 j(也就是原来的 i)指向的元素
            } else {
                i++;
            }
        }
    }
    

    这里要对循环变量 i 进行控制,当删除一个元素时,i 并不能向前进一个位置而应该向后回退一个位置,从这个回退的位置开始重新向前走,才不至于遗漏本应被遍历到的元素。

    代码3:使用迭代器(推荐)

    一种更好的写法是使用迭代器,因为迭代器会自动判断列表中的每一个元素是否被遍历过。事实上,迭代器的实现和代码 2 很接近,但是可读性更好。

    static void remove3(List<Integer> list) {
        Iterator<Integer> iterator = list.iterator();
        while (iterator.hasNext()) {
            Integer next = iterator.next();
            if (next % 2 == 0) {
                iterator.remove();
            }
        }
    }
    

    代码4:使用 removeIf() 方法

    Java 中对 ArrayList 类还支持一种 removeIf() 方法,参数传入一个 Predicate 接口的实现对象或一个 lambda 表达式即可。

    static void remove4(List<Integer> list) {
        list.removeIf(new Predicate<Integer>() {
            @Override
            public boolean test(Integer i) {
                return i % 2 == 0;
            }
        });
    }
    

    或者采用 lambda 表达式,更简洁:

    list.removeIf(i -> i % 2 == 0);
    

    以上就是对列表遍历删除的方法总结。如果条件比较简单,可以直接采用 removeIf() 方法,如果条件比较复杂,那么采用迭代器是一种比较好的方法。

  • 相关阅读:
    HDU 1856 More is better
    并查集模板
    HDU 1325 Is It A Tree?
    HDU 1272 小希的迷宫
    CodeVS 2639 约会计划
    POJ 1163 数字三角形
    HDU 1232 畅通工程
    HDU 1213 How Many Tables
    树形结构打印二叉树
    网址收藏
  • 原文地址:https://www.cnblogs.com/ryuasuka/p/15845660.html
Copyright © 2020-2023  润新知