[Java] 蓝桥杯ALGO-10 算法训练集合运算

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

package algo10;

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

public class Main {
    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();
    }

}

package algo10;

import java.util.Scanner;

import java.util.TreeSet;

public class Main {

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();

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版