Java求多个集合的交集和并集

时间:2022-10-08 17:02:00
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int T = in.nextInt();
int t = T;
while(T--!=0){
int n = in.nextInt();
int i,j;
Integer[][] sz = new Integer[n][];
for(i=0;i<n;i++){
int sizea = in.nextInt();
sz[i] = new Integer[sizea+1];
sz[i][0] = sizea;
for(j=1;j<=sz[i][0];j++)
sz[i][j] = in.nextInt();
}
Set<Integer> set1 = new HashSet<Integer>();
Set<Integer> set2 = new HashSet<Integer>();
Set<Integer> result1 = new HashSet<Integer>();
Set<Integer> result2 = new HashSet<Integer>();
for(i=1;i<=sz[0][0];i++)
set1.add(sz[0][i]);
result1.addAll(set1);
for(i=1;i<n;i++){
for(j=1;j<=sz[i][0];j++){
set2.add(sz[i][j]);
}
//retainAll求两个集合的交集,addAll求两个集合的并集
result1.retainAll(set2);
result2.addAll(set2);
}
//转换成TreeSet自动排序
Set<Integer> re1 = new TreeSet<Integer>(result1);
Set<Integer> re2 = new TreeSet<Integer>(result2);
System.out.println("Case "+(t-T)+":");
for(Integer integer1:re1)
System.out.print(integer1+" ");
System.out.println();
for(Integer integer2:re2)
System.out.print(integer2+" ");
System.out.println();
}
}
}

输入第一行的T表示测试用例的组数,接下来每组测试用例,第一行n表示有n个集合,接下来的n行为具体数据,每一行的第一个数字为该集合的大小,后面的为具体数字,如以下输入表示有1个测试用例,该测试用例有n个集合,第一个集合为{1,2,3,4},第二个集合为{3,4,5,6},……

输出每组测试用例首行为Case #,#代表组数,接下来为交集和并集(都按升序排列)

输入:

1

4

4 1 2 3 4

4 3 4 5 6

3 1 2 3

4 2 1 8 9

输出:

Case 1:

3 4

1 2 3 4 5 6 8 9