前提需要明白 List 是引用类型,引用类型采用引用传递。
我们经常会遇到一些需求求集合的交集、差集、并集。例如下面两个集合:
List<String> list1 = new ArrayList<String>();
list1.add("A");
list1.add("B");
List<String> list2 = new ArrayList<String>();
list2.add("B");
list2.add("C");
求差集
例如,求 List1 中有的但是 List2 中没有的元素:
public static void test3(List list1, List list2) {
list1.removeAll(list2);
System.out.println(list1);
}
结果:
[A]
查看 ArrayList 的 removeAll 的源码
public boolean removeAll(Collection<?> c) {
Objects.requireNonNull(c); return batchRemove(c, false);
}
再查看 batchRemove 的源码:(如果传的第二个参数是 false,保留差集;如果传的是 true,保留的是交集)
private boolean batchRemove(Collection<?> c, boolean complement) {
final Object\[\]elementData = this.elementData;
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) {
System.arraycopy(elementData, r,
elementData, w,
size - r);
w += size - r;
}
if (w != size) { // 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;
}
是重新定义 elementData 数组的元素,下面代码的作用是将本集合中不包含另一个集合的元素重新加入元素,以此实现删除的功能(注意上面调用的方法传的参数是 false,也就是不包含的元素得以保留,实现差集的功能)
if (c.contains(elementData[r]) == complement) {
elementData[w++] = elementData[r];
}
1.求并集(不去重)—-将一个集合全部加入另一个集合
public static void test(List list1, List list2) {
list1.addAll(list2);
System.out.println(list1);
}
结果:
[A, B, B, C]
查看 ArayList 的 addAll()源码是数组复制:
public boolean addAll(Collection<? extends E> c) {
Object[] a = c.toArray();
int numNew = a.length;
ensureCapacityInternal(size + numNew); // Increments modCount
System.arraycopy(a, 0, elementData, size, numNew);
size += numNew;
return numNew != 0;
}
再查看 System 的 arraycopy 发现是一个 native 方法(本地方法):—-其实 system 类的好多方法都是 native 方法
public static native void arraycopy(Object src, int srcPos, Object dest, int destPos, int length);
2.求并集(去重)
例如:求 List1 和 List2 的并集,并实现去重。
思路是:先将 list 中与 list2 重复的去掉,之后将 list2 的元素全部添加进去。
public static void test1(List list1, List list2) {
list1.removeAll(list2);
list1.addAll(list2);
System.out.println(list1);
}
结果:
[A, B, C]
3.求交集
例如:求 List1 和 List2 中都有的元素。
public static void test2(List list1, List list2) {
list1.retainAll(list2);
System.out.println(list1);
}
结果:
[B]
在上面 2 的实验过程中,我们知道 batchRemove(Collection,true)也是求交集,所以猜想 retainAll 内部应该是调用batchRemove(Collection,true),查看 ArrayList 的源码如下:
public boolean retainAll(Collection<?> c) {
Objects.requireNonNull(c); return batchRemove(c, true);
}