Java中使用for循环删除集合中元素需要注意的点

删除集合中的元素,第一反应是遍历集合,比较找到相应的元素然后删除。遍历集合最容易想到的是for循环。

删除集合中为3的元素:

 1 List<Integer> list = new ArrayList<Integer>();
 2         for(int i = 0;i<10;i++){
 3             list.add(i);
 4         }
 5         for(int i = 0;i<10;i++){
 6             list.add(i);
 7         }
 8         System.out.println("删除前"+list);
 9         System.out.println("size:"+list.size());
10         for(int i = 0;i<list.size();i++){
11             if(list.get(i)==3){
12                 list.remove(i);
13             }
14         }
15         System.out.println("删除后"+list);
16         System.out.println("size:"+list.size());

运行结果:

删除前[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
size:20
删除后[0, 1, 2, 4, 5, 6, 7, 8, 9, 0, 1, 2, 4, 5, 6, 7, 8, 9]
size:18

 看起来一点问题也没有,那么我们把集合中元素的顺序换一下。

 1 List<Integer> list = new ArrayList<Integer>();
 2         for(int i = 0;i<10;i++){
 3             list.add(i);
 4             list.add(i);
 5         }
 6         System.out.println("删除前"+list);
 7         System.out.println("size:"+list.size());
 8         for(int i = 0;i<list.size();i++){
 9             if(list.get(i)==3){
10                 list.remove(i);
11             }
12         }
13         System.out.println("删除后"+list);
14         System.out.println("size:"+list.size());

运行结果:

删除前[0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:20
删除后[0, 0, 1, 1, 2, 2, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:19

问题出现了,当删除相邻两个重复元素的时候,只删除一个,这是什么原因呢?

ArrayList的底层结构是数组类型,数组的特点是删除某个元素时,后面所有元素的索引都会往前移,而此时for循环的指针是却是向下移动的。

 

解决方案一:使用每次删除元素的时候,将for循环指针回调一次

 1 List<Integer> list = new ArrayList<Integer>();
 2         for(int i = 0;i<10;i++){
 3             list.add(i);
 4             list.add(i);
 5         }
 6         System.out.println("删除前"+list);
 7         System.out.println("size:"+list.size());
 8         for(int i = 0;i<list.size();i++){
 9             if(list.get(i)==3){
10                 list.remove(i);
11                 i--;
12             }
13         }
14         System.out.println("删除后"+list);
15         System.out.println("size:"+list.size());

运行结果:

删除前[0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:20
删除后[0, 0, 1, 1, 2, 2, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:18

解决方案二:使用迭代器

 1     List<Integer> list = new ArrayList<Integer>();
 2         for(int i = 0;i<10;i++){
 3             list.add(i);
 4             list.add(i);
 5         }
 6         System.out.println("删除前"+list);
 7         System.out.println("size:"+list.size());
 8         Iterator<Integer> it = list.iterator();
 9         while(it.hasNext()){
10             if(3==it.next()){
11                 it.remove();
12             }
13         }
14         System.out.println("删除后"+list);
15         System.out.println("size:"+list.size());

运行结果:

删除前[0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:20
删除后[0, 0, 1, 1, 2, 2, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]
size:18

 为什么使用迭代器可以实现这样的操作呢?下面将基于ArrayList的Iterator的实现分析Iterator的原理。

在ArrayList类中有个方法iterator(),此方法将返回一个iterator的实现,这里可以看出实现类叫Itr,通过其它源码可知,此类是AarryList的内部类,即ArryList的Iterator实现在ArrayList内部

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

ArrayList中实现类Itr类的源码:

 1 private class Itr implements Iterator<E> {
 2         /**
 3          * 下一个返回的位置
 4          */
 5         int cursor = 0;
 6 
 7         /**
 8          * 当前操作的位置
 9          */
10         int lastRet = -1;
11 
12         /**
13          * 类似版本号,检查List是否有更新
14          */
15         int expectedModCount = modCount;
16 
17         public boolean hasNext() {      // 判断是否有下一个元素
18             return cursor != size();
19         }
20 
21         public E next() {  // 返回下一个元素
22             checkForComodification();
23             try {
24                 int i = cursor;     // cursor记录的是下一个元素,所以调用next时将返回的是cursor对应的元素
25                 E next = get(i);    //  记录需要返回的元素
26                 lastRet = i;        // 记录当前元素
27                 cursor = i + 1;     // 记录下一个元素
28                 return next;
29             } catch (IndexOutOfBoundsException e) {
30                 checkForComodification();
31                 throw new NoSuchElementException();
32             }
33         }
34 
35         public void remove() {       // 移除元素
36             if (lastRet < 0)
37                 throw new IllegalStateException();
38             checkForComodification();     //   检查是否有更改,remove或者add
39 
40             try {
41                 AbstractList.this.remove(lastRet);   // 删除当前元素  
42                 if (lastRet < cursor)                // 删除了之后指标减1
43                     cursor--;
44                 lastRet = -1;                    
45                 expectedModCount = modCount;         // 保持版本号一致
46             } catch (IndexOutOfBoundsException e) {
47                 throw new ConcurrentModificationException();
48             }
49         }
50 
51         final void checkForComodification() {    // 如果有更改则抛出ConcurrentModificationException异常
52             if (modCount != expectedModCount)
53                 throw new ConcurrentModificationException();
54         }
55     }

在next()方法中可以看到,通过移动cursor游标来指向集合下一个元素。

再看remove()方法,代码的43行,删除当前元素后,游标前移了。

结论:这两种方法,基本思想都是将指针回调。

posted @ 2019-07-03 10:40  未麻的部屋  阅读(3812)  评论(0编辑  收藏  举报