3.探索ArrayList源码之removeAll和retainAll

  public static  T requireNonNull(T obj) {
     if (obj == null)
         throw new NullPointerException();
     return obj;
 }
  • removeAll
 public boolean removeAll(Collection c) {
     Objects.requireNonNull(c);
     return batchRemove(c, false);
 }
  • retainAll
 public boolean retainAll(Collection c) {
     Objects.requireNonNull(c);
     return batchRemove(c, true);
 }
  • 公共方法
private boolean batchRemove(Collection c, boolean complement) {
    final Object[] elementData = this.elementData;
    // r作为循环次数的纪录
    // w作为保留元素数组的下标
    int r = 0, w = 0;
    // 默认未执行修改操作
    boolean modified = false;
    try {
        for (; r < size; r++)
            if (c.contains(elementData[r]) == complement)
                // 接收需要保留的元素
                elementData[w++] = elementData[r];
    } finally {
        // Preserve behavioral compatibility with AbstractCollection,
        // even if c.contains() throws.
        if (r != size) {    // 表示上面的循环被中断
            // 后面未遍历的元素默认保留,拷贝到elementData要保留的最后一个元素后面
            System.arraycopy(elementData, r,
                             elementData, w,
                             size - r);
            // 因为在上面循环中,通过elementData[w++]这样保存元素
            // 所以跳出循环后,w++,即成了elementData要保留元素的个数
            // 所以,走完上面拷贝方法,最后elementData保留个数等于w + 默认保留的个数
            w += size - r;
        }
        if (w != size) {  // elementData保留个数不等于实际数组大小
             // 说明存在要删除的元素
            // 则从elementData数组的最后要保留的元素后面所有元素置为null
            // 即把要去除的元素置为null,让GC去删除元素
            // clear to let GC do its work
            for (int i = w; i < size; i++)
                elementData[i] = null;
            // 修改的次数即删除了的元素个数
            modCount += size - w;
            //  实际大小就是保留元素的个数
            size = w;
            // 表示修改操作的标志
            modified = true;
        }
    }
    // 表示未修改操作标志
    return modified;
}

解析:

重点就在于它们调用的公共方法
removeAll → return batchRemove(c, false);
retainAll → return batchRemove(c, true);

batchRemove(Collection c, boolean complement)

if (c.contains(elementData[r]) == complement)
           elementData[w++] = elementData[r];
1 在删除的时候,complement为false,表示elementData[r]不在要删除的集合内,则把这些需要保留的元素elementData[r],重置到数组本身(从0开始)elementData[w++]
2 在保留的时候,complement为true,则把这些需要保留的元素elementData[r],重置到数组本身(从0开始)
也就是说,无论是removeAll还是retainAll ,都是将需要保留的元素存起来,最后重置到数组本身。所以通过控制complement为true或者false来操作
换句话说:
当complement为true时,求list中与collection的并集。
当complement为false时,求list与collection的差集。

解析finally后面的方法

r != size; 表示判断r是否会等于size。如果说前面的代码不停止的话,那么最后r会将ArrayList里面的元素全部遍历完毕,最后r的值会等于size的值,并且跳出前面那个for循环。

如果r不等于size:
System.arraycopy(elementData, r,elementData, w, size - r);
表示elementData从r索引开始,截取size-r个元素,然后把截取的所有元素从w开始赋值到elementData
也就是说,如果数组elementData没有遍历完就跳出了循环,把剩下没遍历完的当作不需要删除的数组元素,放到保留的数组元素的数组后面。
如果r等于size:
继续判断,如果w不等于size:
也就是说,原数组elementData已经遍历完,w不等于size表示集合c存在要删除的数组元素
则从w开始遍历,在w往后的元素都置空elementData[i] = null;让GC去工作
8.modCount += size - w;
9.修改数组大小:size = w;
10.modified = true;

你可能感兴趣的:(3.探索ArrayList源码之removeAll和retainAll)