Java算法训练 集合运算
描述
给出两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。
输入
输入描述:
第一行为一个整数n，表示集合A中的元素个数。
第二行有n个互不相同的用空格隔开的整数，表示集合A中的元素。
第三行为一个整数m，表示集合B中的元素个数。
第四行有m个互不相同的用空格隔开的整数，表示集合B中的元素。
集合中的所有元素均为int范围内的整数，n、m<=1000。
输入样例:
5
1 2 3 4 5
5
2 4 6 8 10
输出
输出描述:
第一行按从小到大的顺序输出A、B交集中的所有元素。
第二行按从小到大的顺序输出A、B并集中的所有元素。
第三行按从小到大的顺序输出B在A中的余集中的所有元素。
输出样例:
2 4
1 2 3 4 5 6 8 10
1 3 5
import java.util.Scanner;
import java.util.TreeSet;
public class Main {
public static void main(String[] args) {
TreeSet<Integer> setA = new TreeSet<Integer>();
TreeSet<Integer> setB = new TreeSet<Integer>();
TreeSet<Integer> setAB = new TreeSet<Integer>();
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
for (int i = 0; i < n; i++) {
setA.add(sc.nextInt());
}
setAB.addAll(setA);
int m = sc.nextInt();
for (int i = 0; i < m; i++) {
int temp = sc.nextInt();
if(setA.add(temp)) {//添加失败，说明A集合已经存在该元素，
}else {//直接添加到集合B即可，也就是交集 漂亮！
setB.add(temp);
}
}
setAB.removeAll(setB);//就是将A集合中的B集合元素去掉 即可。
if(setB.size()!=0) {
for(int i:setB) {//B集合为交集
System.out.print(i+" ");
}
System.out.println();
}
if(setA.size()!=0) {
for(int i:setA) {//A集合为并集
System.out.print(i+" ");
}
System.out.println();
}
if(setAB.size()!=0) {
for(int i:setAB) {//AB集合为补集
System.out.print(i+" ");
}
System.out.println();
}
}
}
用Java中的集合去解决集合问题，挺香的。
比C好使多了。
