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+" ");
		}

	}

}