Java实现 蓝桥杯VIP 算法训练 集合运算

问题描述
给出两个整数集合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

第二个可能简短一些

import java.util.Scanner;
import java.util.TreeSet;


public class 集合运算 {
	 public static void main(String[] args) {
	        Scanner in = new Scanner(System.in);
	        
	        TreeSet<Integer> treeSetA = new TreeSet<Integer>();
	        int n = in.nextInt();
	        for (int i = 0; i < n; i++) {
	            treeSetA.add(in.nextInt());
	        }
	        
	        TreeSet<Integer> treeSetB = new TreeSet<Integer>();
	        int m = in.nextInt();
	        for (int i = 0; i < m; i++) {
	            treeSetB.add(in.nextInt());
	        }
	        in.close();
	        
	        printDivide(treeSetA, treeSetB);
	        printAdd(treeSetA, treeSetB);
	        printRemain(treeSetA, treeSetB);
	    }
	    
	    public static void printAdd(TreeSet<Integer> a, TreeSet<Integer> b) {
	        a.addAll(b);
	        for (int i : a) {
	            System.out.print(i + " ");
	        }
	        System.out.println();
	    }
	    
	    public static void printDivide(TreeSet<Integer> a, TreeSet<Integer> b) {
	        
	        @SuppressWarnings("unchecked")
	        TreeSet<Integer> tempA = (TreeSet<Integer>) a.clone();
	        a.removeAll(b);
	        tempA.removeAll(a);
	        for (int i : tempA) {
	            System.out.print(i + " ");
	        }
	        System.out.println();
	    }
	    
	    public static void printRemain(TreeSet<Integer> a, TreeSet<Integer> b) {
	        a.removeAll(b);
	        for (int i : a) {
	            System.out.print(i + " ");
	        }
	        System.out.println();
	    }


}

import java.util.Scanner;
import java.util.TreeSet;

public class Demo1两个集合 {
	public static void main(String[] args) {
		TreeSet<Integer> seta = new  TreeSet<Integer>();
		TreeSet<Integer> setb = new  TreeSet<Integer>();
		TreeSet<Integer> setab = new  TreeSet<Integer>();
		 
		int temp=0;
		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++) {
			temp=sc.nextInt();
			if(seta.add(temp)){
				
			}else{
				setb.add(temp);
			}
		}
		sc.close();
		setab.removeAll(setb);
		for(int i:setb){
			System.out.print(i+" ");
		}
		if(setb.size()!=0)
		System.out.println();
		for(int i:seta){
			System.out.print(i+" ");
		}
		if(seta.size()!=0)
		System.out.println();
		for(int i:setab){
			System.out.print(i+" ");
		}
		 
				
	}

}