java蓝桥杯练习 集合运算

java蓝桥杯练习 集合运算

资源限制
时间限制：1.0s 内存限制：512.0MB
问题描述
　　给出两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。
输入格式
　　第一行为一个整数n，表示集合A中的元素个数。
　　第二行有n个互不相同的用空格隔开的整数，表示集合A中的元素。
　　第三行为一个整数m，表示集合B中的元素个数。
　　第四行有m个互不相同的用空格隔开的整数，表示集合B中的元素。
　　集合中的所有元素均为int范围内的整数，n、m<=1000。
输出格式
　　第一行按从小到大的顺序输出A、B交集中的所有元素。
　　第二行按从小到大的顺序输出A、B并集中的所有元素。
　　第三行按从小到大的顺序输出B在A中的余集中的所有元素。
样例输入
5
1 2 3 4 5
5
2 4 6 8 10
样例输出
2 4
1 2 3 4 5 6 8 10
1 3 5
样例输入
4
1 2 3 4
3
5 6 7
样例输出
1 2 3 4 5 6 7
1 2 3 4

//java code

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n=sc.nextInt();
        TreeSet<Integer> treeA=new TreeSet<Integer>();
        for (int i = 0; i < n; i++) {
            treeA.add(sc.nextInt());
        }
        int m=sc.nextInt();
        TreeSet<Integer> treeB=new TreeSet<Integer>();
        for (int i = 0; i < m; i++) {
            treeB.add(sc.nextInt());
        }
        printAdd(treeA,treeB);
        printd(treeA,treeB);
        printRemain(treeA,treeB);
    }
    static void printd(TreeSet<Integer> a,TreeSet<Integer> b){
        a.addAll(b);
        for (int i:
             a) {
            System.out.print(i+" ");
        }
        System.out.println();
    }
    static void printAdd(TreeSet<Integer> a,TreeSet<Integer> b){
        TreeSet<Integer> temp=(TreeSet<Integer>)a.clone();
        a.removeAll(b);
        temp.removeAll(a);
        for (int i:temp
             ) {
            System.out.print(i+" ");
        }
        System.out.println();
    }
    static void printRemain(TreeSet<Integer> a,TreeSet<Integer> b){
        a.removeAll(b);
        for (int i:
             a) {
            System.out.print(i+" ");
        }
        System.out.println();
    }
}