集合运算java算法_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 treeSetA = new TreeSet();

int n = in.nextInt();

for (int i = 0; i < n; i++) {

treeSetA.add(in.nextInt());

}

TreeSet treeSetB = new TreeSet();

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 a, TreeSet b) {

a.addAll(b);

for (int i : a) {

System.out.print(i + " ");

}

System.out.println();

}

public static void printDivide(TreeSet a, TreeSet b) {

@SuppressWarnings("unchecked")

TreeSet tempA = (TreeSet) a.clone();

a.removeAll(b);

tempA.removeAll(a);

for (int i : tempA) {

System.out.print(i + " ");

}

System.out.println();

}

public static void printRemain(TreeSet a, TreeSet 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 seta = new TreeSet();

TreeSet setb = new TreeSet();

TreeSet setab = new TreeSet();

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

}

}

}