首页 > 代码库 > 【Java】两个ArrayList之间求交并补
【Java】两个ArrayList之间求交并补
同样的方法应该也使用在《【Java】Java中的Collections类——Java中升级版的数据结构》(点击打开链接)中提及到Java中的各类集合,这里拿各位编程最常用的ArrayList做例子。这个东西除了用来做可变形数组以外,有时候还会出现要在两个ArrayList之间进行集合运算,我在这里举出最常见的求交并补的例子,其余的复杂的集合运算,请自己打开《离散数学》或者《数理逻辑》一书慢慢推导吧。我们是程序猿,不是数学家,要在两个ArrayList之间求交并补已经很少见了,何况其它……
这里的交并补不需要通过for循环等实现,这里类本身封装好方法。
1、首先是最简单的交集,用一个已经封装好的retainAll就解决问题了:
比如以下的语句段,设置一个{1,2,3}与{3,4}的ArrayList求交:
ArrayList<Integer> arraylist1 = new ArrayList<Integer>(); ArrayList<Integer> arraylist2 = new ArrayList<Integer>(); arraylist1.add(1);arraylist1.add(2);arraylist1.add(3); arraylist2.add(3);arraylist2.add(4); arraylist1.retainAll(arraylist2); System.out.println("Arraylist1∩Arraylist2="+arraylist1);就可以得到如下的运行结果:
Arraylist1∩Arraylist2=[3]
这里值得注意的是,不要写成:arraylist1=arraylist1.retainAll(arraylist2);,arraylist1.retainAll(arraylist2);本身就完成arraylist1与arraylist2求交,并且更新arraylist1的操作,arraylist1.retainAll(arraylist2);执行成功则返回一个true,当然,我就从来没见过它执行失败,返回false的。arraylist1是一个存放整形的ArrayList<Integer>,根本就不能相等,这样写Eclipse绝对会报错。这里求交集arraylist1与arraylist2的顺序没有问题的,只是求完之后更新哪个arraylist的问题。
2、之后是求补集,也是用一个封装好的removeAll就解决问题:
比如以下的语句段,设置一个{3}对于全集{1,2,3}求补:
ArrayList<Integer> arraylist1 = new ArrayList<Integer>(); ArrayList<Integer> arraylist2 = new ArrayList<Integer>(); arraylist1.add(1);arraylist1.add(2);arraylist1.add(3); arraylist2.add(3); arraylist1.removeAll(arraylist2); System.out.println("┐Arraylist2(对于Arraylist1)="+arraylist1);运行结果如下:
┐Arraylist2(对于Arraylist1)=[1, 2]
这里要注意的问题同上面的交集,如果你这样写,以下是错误示范:
ArrayList<Integer> arraylist1 = new ArrayList<Integer>(); ArrayList<Integer> arraylist2 = new ArrayList<Integer>(); arraylist1.add(1);arraylist1.add(2);arraylist1.add(3); arraylist2.add(3); System.out.println("┐Arraylist2(对于Arraylist1)"+arraylist1.removeAll(arraylist2));那么则得到下面的运行结果:
┐Arraylist2(对于Arraylist1)true
而不是┐Arraylist2(对于Arraylist1)=[1, 2],原因同交集
3、并集则不同,这里没有封装好方法,我们写一个方法,把要求并的两个ArrayList的所有元素叠加在一起,再去重,去重可以利用上次我在《【Java】为ArrayList去重》(点击打开链接)介绍的方法
比如:
import java.util.*; class ArraylistCalculate{ // 两个整数集求并集 public ArrayList<Integer> integerArrayListUnion( ArrayList<Integer> arraylist1, ArrayList<Integer> arraylist2) { ArrayList<Integer> arraylist = new ArrayList<Integer>(); arraylist.addAll(arraylist1); arraylist.addAll(arraylist2); arraylist = new ArrayList<Integer>(new HashSet<Integer>(arraylist)); return arraylist; } } public class JavaArraylistCalculate { public static void main(String args[]){ ArrayList<Integer> arraylist1 = new ArrayList<Integer>(); ArrayList<Integer> arraylist2 = new ArrayList<Integer>(); arraylist1.add(1);arraylist1.add(2);arraylist1.add(3); arraylist2.add(3);arraylist2.add(4); System.out.println("Arraylist1∪Arraylist2="+new ArraylistCalculate().integerArrayListUnion(arraylist1, arraylist2)); } }
运行结果如下:
Arraylist1∪Arraylist2=[1, 2, 3, 4]
当然,我们也可以把上述介绍得两个ArrayList之间求交并补的所有方法封装起来,得到如下的类:
class ArraylistCalculate{ // 两个整数集求差集 public ArrayList<Integer> integerArrayListDifference( ArrayList<Integer> arraylist1, ArrayList<Integer> arraylist2) { arraylist1.removeAll(arraylist2); return arraylist1; } // 两个整数集求并集 public ArrayList<Integer> integerArrayListUnion( ArrayList<Integer> arraylist1, ArrayList<Integer> arraylist2) { ArrayList<Integer> arraylist = new ArrayList<Integer>(); arraylist.addAll(arraylist1); arraylist.addAll(arraylist2); arraylist = new ArrayList<Integer>(new HashSet<Integer>(arraylist)); return arraylist; } // 两个整数集求交集 public ArrayList<Integer> integerArrayListIntersections( ArrayList<Integer> arraylist1, ArrayList<Integer> arraylist2) { arraylist1.retainAll(arraylist2); return arraylist1; } }
需要的时候调用就可以了。
当然,由于用到了ArrayList请注意在开头引用import java.util.*;
【Java】两个ArrayList之间求交并补