蓝桥杯 试题 算法训练 集合运算（java）

问题描述

　　输入两个整数集合A、B，求出他们的交集、并集以及B在A中的余集。交集、并集和余集的计算都要求写成一个单独的函数。
　　输入第一行为一个正整数n，表示集合A中的元素个数；第二行有n个按从小到大的顺序输入且互不相同的整数，表示集合A中的元素；第三行为一个正整数m，表示集合B中的元素个数；第四行有m个按从小到大的顺序输入且互不相同的整数，表示集合B中的元素；集合中的所有元素均为int范围内的整数，1≤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

public class AL_492 {
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int a=sc.nextInt();
        int[] arr=new int[a];
        for (int i = 0; i < arr.length; i++) {
            arr[i]=sc.nextInt();
        }
        int b=sc.nextInt();
        int[] abb=new int[b];
        for (int i = 0; i < abb.length; i++) {
            abb[i]=sc.nextInt();
        }
        //交集
        for (int i = 0; i < a; i++) {
            for (int j = 0; j < b; j++) {
                if (arr[i]==abb[j]){
                    //逐个比较，相等即为公共的数
                    System.out.print(arr[i]+" ");
                }
            }
        }
        System.out.println();//换行
        //并集
        Set<Integer> s=new TreeSet<>();
        //HashSet是无序的，而TreeSet是升序的，更符合题目要求
        for (int i = 0; i < a; i++) {
            s.add(arr[i]);
        }for (int i = 0; i < b; i++) {
            s.add(abb[i]);
        }
        s.forEach(x-> System.out.print(x+" "));//遍历输出
        System.out.println();//换行
        //余集
        for (int i=0;i<arr.length;i++){
            int n=0;
            while (n<abb.length){
                if (arr[i]==abb[n]) break;
                if (n==abb.length-1){//不相同的逐个比较
                    System.out.print(arr[i]+" ");
                }
                n+=1;
            }
        }
    }
}