是这样的,我想判断集合A中的元素在集合B中是否出现的,用for循环去遍历,显然代码比较臃肿,而且效率也比较低下,所以我想到了求两个集合的交集就行。在Java中,求两个集合的交集使用的是retainAll()方法,他会返回一个布尔值。
如下我创建了两个集合
@Test
public void test(){
List<String> a= Arrays.asList("1","2","3");
List<String> b= Arrays.asList("1","2");
boolean contain = a.retainAll(b);
System.out.println("包含:"+contain);
}
当我执行上面语句的时候,竟然报了一个错:
但是换成b.retainAll(a) 却不报任何错误
输出结果如下:
而且当两个集合完素完全相同的时候,返回值也为false,只有不相同的时候才为true,我有点奇怪了,两个集合完全相同,不应该是为true的嘛。
后来我查阅资料才知道这个方法的用途:
retainAll() 方法用于保留 arraylist 中在指定集合中也存在的那些元素,也就是删除指定集合中不存在的那些元素。
list1与list2做交集,结果集与list2做比较,如果相同返回true,否则返回false
retain是保留的意思,list1与list2做交集,结果集赋值给list1,如果list1被改变返回true,否则返回false
也就是说当我执行a.reatinAll(b),集合A会保留与集合B中相同的元素,去除不相同的元素。那这样集合是改变了,应该是为ture,为啥会抛出异常咯。于是追踪源码:
AbstractCollection
可以看到集合A中的元素不在集合B中,就会执行remove方法。继续进入remvoe方法:
我们这个方法会默认抛出异常。
但是我之前测试的是正常的,这就很奇怪了,于是我就猜测可能是因为我创建的集合有问题叭,于是我就改为创建一个ArrayList集合,然后List接口接收。
List<String> a=new ArrayList<>();
a.add("1");
a.add("2");
a.add("3");
List<String> b=new ArrayList<>();
b.add("1");
b.add("2");
boolean contain = a.retainAll(b);
System.out.println("包含:"+contain);
再次运行这段代码的时候,发现返回结果为true。
我们再来追踪下源代码:
我们可以看到这次调用的是ArrayList的retainAll方法。首先它会调用一个requireNonNull()方法,简明思议就是判断我们传入的集合是否为空,我们可以进入去看下:
不为空的话,再调用了一个batchRemove方法,可以看到这个方法传了两个参数,一个是我们传入的参数,一个是布尔值,默认为true.我们再进入这个方法里面看下:
我们进入去会发现,它首先是将我们集合A中的元素放到了一个空数组中。
接下来就是判断集合A中的元素是否在集合B中,如果在的话,还是重新赋值给我们创建的数组中。
最后执将r和w与集合A的大小进行比较
我们可以看到当r不等于集合A的大小时,会执行一个数组的拷贝,也就是相同的数据重新赋值给集合A。如果w不等于集合A的大小时。会将不相同的位置索引处置为空,并将modified赋值为true并返回。此时我们可以打印下,执行retainAll方法后,集合A和集合B中的元素。
这里可以看到集合A只保留了在集合B中也存在的元素。
总结:
1.arrayList.retainAll(Collection c),是保留集合c中相同的元素,并重新赋值个给arrayList,要弄清楚主次关系。我们要判断一个集合中的元素是否在另一个集合中也出现,那么这个需要判断的集合就要作为参数。
2.arrayList必须是一个ArrayList类对象
3.我们需要判断的集合不能为空
除了这种方法我们还可以使用stream流来就交集的。
List<String> intersection = a.stream().filter(item -> b.contains(item)).collect(Collectors.toList());