public static <T> 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;