正确在遍历中删除List元素

最近在写代码的时候遇到了遍历时删除List元素的问题，在此写一篇博客记录一下。
一般而言，遍历List元素有以下三种方式：

• 使用普通for循环遍历
• 使用增强型for循环遍历
• 使用iterator遍历

使用普通for循环遍历

代码如下：

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new ArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		for (int i = 0; i < list.size(); i++) {
			// index and number
			System.out.print(i + " " + list.get(i));
			if (list.get(i) % 2 == 0) {
				list.remove(list.get(i));
				System.out.print(" delete");
				i--; // 索引改变!
			}
			System.out.println();
		}
	}
}

结果如下：

可以看到遍历删除偶数的结果是成功的，但是这种方法由于删除的时候会改变list的index索引和size大小，可能会在遍历时导致一些访问越界的问题，因此不是特别推荐。

使用增强型for循环遍历

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new ArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		for (Integer num : list) {
			// index and number
			System.out.print(num);
			if (num % 2 == 0) {
				list.remove(num);
				System.out.print(" delete");
			}
			System.out.println();
		}
	}
}

结果如下：

可以看到删除第一个元素时是没有问题的，但删除后继续执行遍历过程的话就会抛出ConcurrentModificationException的异常。

使用iterator遍历

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new ArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		Iterator<Integer> it = list.iterator();
		while (it.hasNext()) {
			// index and number
			int num = it.next();
			System.out.print(num);
			if (num % 2 == 0) {
				it.remove();
				System.out.print(" delete");
			}
			System.out.println();
		}
	}
}

结果如下：

可以看到顺利的执行了遍历并删除的操作，因此最推荐的做法是使用iterator执行遍历删除操作。

以上是关于非线程安全的ArrayList，如果是线程安全的CopyOnWriteArrayList呢？

使用普通for循环遍历

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new CopyOnWriteArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		for (int i = 0; i < list.size(); i++) {
			// index and number
			System.out.print(i + " " + list.get(i));
			if (list.get(i) % 2 == 0) {
				list.remove(list.get(i));
				System.out.print(" delete");
				i--; // 索引改变!
			}
			System.out.println();
		}
	}
}

结果如下：

可以看到遍历删除是成功的，但是这种方法由于删除的时候会改变list的index索引和size大小，可能会在遍历时导致一些访问越界的问题，因此不是特别推荐。

使用增强型for循环遍历

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new CopyOnWriteArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		for (Integer num : list) {
			// index and number
			System.out.print(num);
			if (num % 2 == 0) {
				list.remove(num);
				System.out.print(" delete");
			}
			System.out.println();
		}
	}
}

结果如下：

可以看见与ArrayList遍历删除时情况不同，CopyOnWriteArrayList是允许使用增强型for进行循环遍历删除的。

使用iterator遍历

public class Main {
	public static void main(String[] args) throws Exception {
		List<Integer> list = new CopyOnWriteArrayList<>();
		for (int i = 0; i < 5; i++)
			list.add(i);
		// list {0, 1, 2, 3, 4}
		Iterator<Integer> it = list.iterator();
		while (it.hasNext()) {
			// index and number
			int num = it.next();
			System.out.print(num);
			if (num % 2 == 0) {
				it.remove();
				System.out.print(" delete");
			}
			System.out.println();
		}
	}
}

结果如下：

与ArrayList不同，由于CopyOnWriteArrayList的iterator是对其List的一个“快照”，因此是不可改变的，所以无法使用iterator遍历删除。

综上所述，当使用ArrayList时，我们可以使用iterator实现遍历删除；而当我们使用CopyOnWriteArrayList时，我们直接使用增强型for循环遍历删除即可，此时使用iterator遍历删除反而会出现问题。