当前位置:   article > 正文

ArrayList循环遍历并删除元素的常见陷阱

arraylist循环遍历并删除元素的常见陷阱

在工作和学习中,经常碰到删除ArrayList里面的某个元素,看似一个很简单的问题,首先请看下面的例子:

  1. public class DuplicateAndOrderTest{
  2. public static void main(String[] args){
  3. ArrayList<String>list=new ArrayList<>();
  4. list.add("a");
  5. list.add("b");
  6. list.add("b");
  7. list.add("c");
  8. list.add("c");
  9. list.add("c");
  10. remove(list);
  11. for(String s:list){
  12. System.out.println("element : "+s);
  13. }
  14. }
  15. private static void remove(ArrayList<String> list) {
  16. }
  17. }

如果要想删除list的b字符,有下面两种常见的错误例子:

错误写法实例一: 这种最普通的循环写法执行后会发现第二个“b”的字符串没有删掉:

  1. private static void remove(ArrayList<String> list) {
  2. for (int i = 0; i < list.size(); i++) {
  3. String s = list.get(i);
  4. if (s.equals("b")) {
  5. list.remove(s);
  6. }
  7. }
  8. }

实例一的错误原因。翻开JDK的ArrayList源码,先看下ArrayList中的remove方法(注意ArrayList中的remove有两个同名方法,只是入参不同,这里看的是入参为Object的remove方法)是怎么实现的: 

  1. public boolean remove(Object o) {
  2. if (o == null) {
  3. for (int index = 0; index < size; index++)
  4. if (elementData[index] == null) {
  5. fastRemove(index);
  6. return true;
  7. }
  8. } else {
  9. for (int index = 0; index < size; index++)
  10. if (o.equals(elementData[index])) {
  11. fastRemove(index);
  12. return true;
  13. }
  14. }
  15. return false;
  16. }
  17. 一般情况下程序的执行路径会走到else路径下最终调用faseRemove方法
  18. private void fastRemove(int index) {
  19. modCount++;
  20. int numMoved = size - index - 1;
  21. if (numMoved > 0)
  22. System.arraycopy(elementData, index+1, elementData, index,
  23. numMoved);
  24. elementData[--size] = null; // clear to let GC do its work
  25. }

可以看到会执行System.arraycopy方法,导致删除元素时涉及到数组元素的移动。针对错误写法一,在遍历第一个字符串b时因为符合删除条件,所以将该元素从数组中删除,并且将后一个元素移动(也就是第二个字符串b)至当前位置,导致下一次循环遍历时后一个字符串b并没有遍历到(要删除的元素相邻会有这个问题 !!!),所以无法删除。针对这种情况可以倒序删除的方式来避免,因为数组倒序遍历时即使发生元素删除也不影响后序元素遍历:

  1. private static void remove(ArrayList<String> list) {
  2. for (int i = list.size() - 1; i >= 0; i--) {
  3. String s = list.get(i);
  4. if (s.equals("b")) {
  5. list.remove(s);
  6. }
  7. }
  8. }

 

错误写法实例二:这种for-each写法会报出著名的并发修改异常:java.util.ConcurrentModificationException

  1. private static void remove(ArrayList<String> list) {
  2. for (String s : list) {
  3. if (s.equals("b")) {
  4. list.remove(s);
  5. }
  6. }
  7. }

错误二产生的原因却是foreach写法是对实际的Iterable、hasNext、next方法的简写,

  1. for (String s : list) {
  2. System.out.println(s);
  3. }
  4. 增强for循环底层 经过反编译后得到如下
  5. String s;
  6. for(Iterator iterator = list.iterator(); iterator.hasNext(); System.out.println(s)){
  7. s = (String)iterator.next();
  8. }

问题同样处在上文的fastRemove方法中,可以看到第一行把modCount变量的值加一,而expectedCount值未变,但在ArrayList返回的迭代器(该代码在其父类AbstractList中):

  1. public Iterator<E> iterator() {
  2. return new Itr();
  3. }

这里返回的是AbstractList类内部的迭代器实现private class Itr implements Iterator,看这个类的next方法:

  1. public E next() {
  2. checkForComodification();
  3. int i = cursor;
  4. if (i >= size)
  5. throw new NoSuchElementException();
  6. Object[] elementData = ArrayList.this.elementData;
  7. if (i >= elementData.length)
  8. throw new ConcurrentModificationException();
  9. cursor = i + 1;
  10. return (E) elementData[lastRet = i];
  11. }
  12. 这里会做迭代器内部修改次数检查,因为上面的remove(Object)方法修改了modCount的值,而expectedCount值未变, 所以才会报出并发修改异常
  13. final void checkForComodification() {
  14. if (modCount != expectedModCount)
  15. throw new ConcurrentModificationException();
  16. }

List 移除某个元素的四种正确方式

方式一,使用 Iterator ,顺序向下,如果找到元素,则使用 remove 方法进行移除。
方式二,倒序遍历 List ,如果找到元素,则使用 remove 方法进行移除。
方式三,正序遍历 List ,如果找到元素,则使用 remove 方法进行移除,然后进行索引 “自减”。
方式四,使用jdk1.8新增的Stream流操作

 

1.Iterator 迭代器

要避免这种情况的出现则在使用迭代器迭代时(显示或for-each的隐式)不要使用ArrayList的remove,改为用Iterator的remove即可。

  1. private static void remove(ArrayList<String> list) {
  2. Iterator<String> it = list.iterator();
  3. while (it.hasNext()) {
  4. String s = it.next();
  5. if (s.equals("b")) {
  6. it.remove();
  7. }
  8. }
  9. }

2 倒序遍历

当我们倒序遍历元素的时候,无论删除元素之后的元素怎么移动,之前的元素对应的索引(index)是不会发生变化的,所以在删除元素的时候不会发生问题。

  1. private static void remove(ArrayList<String> list) {
  2. for (int i = list.size() - 1; i > 0; i--) {
  3. if("b".equals(list.get(i))){
  4. list.remove(i);
  5. }
  6. }
  7. }

3.正序遍历

  1. private static void remove(ArrayList<String> list) {
  2. for (int i = 0; i < list.size(); i++) {
  3. if("b".equals(list.get(i))){
  4. list.remove(i);
  5. i--;
  6. }
  7. }
  8. }

4. jdk 1.8新特性

list.stream().filter(s-> !s.equals("b")).collect(Collectors.toList());

原文: https://www.cnblogs.com/huangjinyong/p/9455163.html

          https://blog.csdn.net/zhaozuhao110/article/details/88116831

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/470410
推荐阅读
  

闽ICP备14008679号