java集合如何删除指定元素

首先是一个最容易想到的实现：

@Test
public void test28() {
    List<String> demos = new ArrayList<>();
    demos.add("abc");
    demos.add("cde");
    demos.add("cde");
    for (String demo : demos) {
        if (demo.equals("cde")) {
            demos.remove("cde");
        }
    }
    System.out.println(JSON.toJSON(demos));

    //    结果：["abc","cde"]
}

循环删除，可惜结果并不对，可以看到这种方式只能删除一个指定元素，如果指定元素有多个，就不适用了。

显然底层是删除一个元素后，后续的元素都往前推，数组下标都变了。

比如删除下标为1的元素，删完后，原来下标为2的元素下边变为了1，但是检测元素的指针下标还是2，导致漏掉了一个。

试试集合提供的方法呢：

@Test
public void test29() {
    List<String> demos = new ArrayList<>();
    demos.add("abc");
    demos.add("cde");
    demos.add("cde");
    demos.remove("cde");
    System.out.println(JSON.toJSON(demos));

    //    结果：["abc","cde"]
}

和自己单次循环删除的效果一样，保不齐底层实现都一样。。

既然一次只能删一个，那循环多次是否就能保证全删掉呢：

@Test
public void test30() {
    List<String> demos = new ArrayList<>();
    demos.add("abc");
    demos.add("cde");
    demos.add("cde");
    int i = 0;
    while (i++ < demos.size())
        for (String demo : demos) {
            if (demo.equals("cde")) {
                demos.remove("cde");
            }
            System.out.println(i);
        }
    System.out.println(JSON.toJSON(demos));

    //    1
    //1
    //2
    //2
    //
    //java.util.ConcurrentModificationException
    // at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:909)
    // at java.util.ArrayList$Itr.next(ArrayList.java:859)
}

直接报异常了，看样子是并发的问题

再试试迭代器：

@Test
public void test31() {
    List<String> demos = new ArrayList<>();
    demos.add("abc");
    demos.add("cde");
    demos.add("cde");
    Iterator<String> iterator = demos.iterator();
    while (iterator.hasNext()) {
        String next = iterator.next();
        if (next.equals("cde")) {
            iterator.remove();
        }
    }
    System.out.println(JSON.toJSON(demos));
    //    结果：
    //    ["abc"]
}

可以，迭代器成功删除了所有指定元素

再试试stream的api

@Test
public void test32() {
    List<String> demos = new ArrayList<>();
    demos.add("abc");
    demos.add("cde");
    demos.add("cde");
    demos.removeIf(next -> next.equals("cde"));
    System.out.println(JSON.toJSON(demos));
    // 结果
    //    ["abc"]
}

其实这种写法和上面的迭代器一样，但是更优雅，也同样实现了。推荐。