首页 > 代码库 > [ALGO-10] 集合运算

[ALGO-10] 集合运算

算法训练 集合运算  
时间限制: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
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);

		while (scanner.hasNext()) {
			int n = scanner.nextInt();
			List<Integer> A = new ArrayList<>();
			for (int i = 0; i < n; i++) {
				A.add(scanner.nextInt());
			}

			int m = scanner.nextInt();
			List<Integer> B = new ArrayList<>();
			for (int i = 0; i < m; i++) {
				B.add(scanner.nextInt());
			}

			List<Integer> U = new ArrayList<>();
			U.addAll(B);

			B.retainAll(A);
			A.removeAll(B);

			U.addAll(A);

			Collections.sort(U);
			Collections.sort(A);
			Collections.sort(B);

			for (int i = 0; i < B.size(); i++) {
				System.out.print(B.get(i));
				System.out.print(i == B.size() - 1 ? "\r\n" : " ");
			}

			for (int i = 0; i < U.size(); i++) {
				System.out.print(U.get(i));
				System.out.print(i == U.size() - 1 ? "\r\n" : " ");
			}

			for (int i = 0; i < A.size(); i++) {
				System.out.print(A.get(i));
				System.out.print(i == A.size() - 1 ? "\r\n" : " ");
			}
		}
	}
}