Java中如何优雅地删除List中的元素

在工作中的许多场景下，我们都会使用到List这个数据结构，那么同样的有很多场景下需要删除List中的某一个元素或某几个元素，那么我们该如何正确无误地删除List中的元素的，今天我来教大家三种方式。

前提知识准备

for循环的执行顺序

这里借用百度百科的一张图，简明扼要的介绍一下。

Iterator迭代器介绍

迭代器：迭代其实我们可以简单地理解为遍历，是一个标准化遍历各类容器里面的所有对象的方法类，它是一个很典型的设计模式。Iterator 模式是用于遍历集合类的标准访问方法。它可以把访问逻辑从不同类型的集合类中抽象出来，从而避免向每次遍历前都需要知道要遍历集合的内部结构。 

// 遍历list
List list = new ArrayList();
list.add(1);
list.add(2);
for (int i = 0; i < list.size(); i++) {
    Object object = list.get(i);
    // do something
}

// 遍历map
Map<String,String> map = new HashMap<>();
map.put("1","first");
map.put("2","second");
for (Map.Entry<String,String> entry : map.entrySet()){
    String key = entry.getKey();
    String value = entry.getValue();
    // do something
}

对于这两种方式，我们总是都事先知道集合的内部结构，访问代码和集合本身是紧密耦合的，无法将访问逻辑从集合类和遍历方法中分离出来。同时每一种集合对应一种遍历方法，代码无法复用。
为了解决以上问题， Iterator 模式腾空出世，它总是用同一种逻辑来遍历集合。使得需要遍历集合的人，在遍历的时候不需要了解集合的内部结构，所有的内部状态都由 Iterator 来维护。遍历集合的方法不直接和集合类打交道，它总是控制 Iterator，向它发送”向前”，”向后”，”取当前元素”的命令，就可以间接遍历整个集合。

错误：for循环顺序遍历

直接使用简单for循环，以for (int i = 0; i < list.size(); i++) 进行遍历，这种方式可能会在遍历的过程中漏掉部分元素，从而出现少删的情况。

/**
 * 通过简单的遍历方式，在遍历的过程中有可能会漏掉元素
 * 取第二个元素i=1时，满足条件被删掉，原有的数组的第三个元素，变成了新数组的第二个元素
 * i++后i=2,但i=2指向的是新数组中的第三个元素，那么原数组中的第三个元素就被漏掉了
 *
 * @param list
 * @param element
 * @return
 */
public static List forRemove(List list, Object element) {
    for (int i = 0; i < list.size(); i++) {
        if (element.equals(list.get(i))) {
            list.remove(i);
        }
    }
    return list;
}

错误：增强for循环，删除后不退出

使用增强for循环是，如果删除后继续向下循环则会报java.util.ConcurrentModificationException

/**
 * 使用增强for循环是，如果删除后继续向下循环则会报
 * java.util.ConcurrentModificationException
 *
 * @param list
 * @param element
 * @return
 */
public static List forceForRemove(List list, Object element) {
    for (Object item : list) {
        if (item.equals(element)) {
            list.remove(item);
        }
    }
    return list;
}

异常如下：

Exception in thread "main" java.util.ConcurrentModificationException
    at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:901)
    at java.util.ArrayList$Itr.next(ArrayList.java:851)
    at com.lingyejun.leetcode.RemoveListElement.forceForRemove(RemoveListElement.java:57)
    at com.lingyejun.leetcode.RemoveListElement.main(RemoveListElement.java:112)

正确：逆向循环遍历

我们使用逆向遍历的方式可以得到正确的结果

/**
 * 逆向循环，是正确的
 * 1-->2-->3-->4
 * 逆向循环时，倒数第一个元素满足条件被删除时，i--后,原数组的倒数第二个变成了新数组的倒数第一个元素
 * i = size-2指向新数组的最后一个元素，没有漏掉。
 * 同理倒数第二个元素满足条件被删除时，i--后,原数组的倒数第三个变成了新数组的倒数第二个元素
 * i= size-3指向新数组的倒数第二个元素，也没有漏掉
 *
 * @param list
 * @param element
 * @return
 */
public static List reverseorRemove(List list, Object element) {
    for (int i = list.size() - 1; i >= 0; i--) {
        if (element.equals(list.get(i))) {
            list.remove(i);
        }
    }
    return list;
}

正确但不优雅：增强for循环删除元素后break

使用增强for循环，删除元素后，立即跳出，则正常退出，但缺点是不能向后继续循环了。　　

/**
 * 删除元素后，立即跳出，则正常退出，但不能向后继续循环了
 *
 * @param list
 * @param element
 * @return
 */
public static List forceForRemove1(List list, Object element) {
    for (Object item : list) {
        if (item.equals(element)) {
            // 删除后立马终端循环，会正常跳出，但代价是不能继续向后循环了
            list.remove(item);
            break;
        }
    }
    return list;
}

优雅删除：使用Iterator迭代器

使用迭代器可，正确无误的删除，代码简洁优雅，推荐使用！

 

/**
 * 使用迭代器可，正确无误的删除
 * 
 * @param list
 * @param element
 * @return
 */
public static List iteratorRemove(List list, Object element) {
    Iterator iterator = list.iterator();
    while (iterator.hasNext()) {
        Object cur = iterator.next();
        if (cur.equals(element)) {
            // 注意！！！这里时Iterator.remove()!!!而不是list.remove()!!!
            iterator.remove();
        }
    }
    return list;
}

 

参考文章：

https://jingyan.baidu.com/article/7f766dafaa6ee04101e1d0e6.html

http://wiki.jikexueyuan.com/project/java-enhancement/java-thirty.html