JAVA蓝桥杯算法训练集合运算

问题描述
  给出两个整数集合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.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner read = new Scanner(System.in);
		Map<Integer,Integer> map = new HashMap<Integer,Integer>();//用HashMap来存储A中的数字,找的时候可以省时间
		List<Integer> list = new ArrayList<Integer>();
		int n = read.nextInt();
		int num;
		int []numN = new int[n];
		for(int i=0;i<n;i++)
		{
			num = read.nextInt();
			numN[i] = num;
			map.put(i, num);
			list.add(num);
		}
		int m = read.nextInt();
		int []numM = new int[m];
		for(int i=0;i<m;i++)
		{
			num = read.nextInt();
			numM[i] = num;
			if(list.contains(Integer.valueOf(num)))
			{
				list.remove(Integer.valueOf(num));
			}
		}
		//此时A中的所有元素都在map中了,输入也完成了
		int []numJiao = new int[m+n];//记录交集
		int countJiao = 0;//交集中有几个元素
		int []numBing = new int[m+n];//并集
		int countBing = 0;
		for(int i=0;i<n;i++)
		{
			numBing[i] = numN[i];
			countBing++;
		}
		for(int i=0;i<numM.length;i++)
		{
			if(map.containsValue(numM[i]))
			{
				numJiao[countJiao] = numM[i];
				countJiao++;
			}
			else
			{
				numBing[countBing] = numM[i];
				countBing++;
			}
		}
		
		Arrays.sort(numBing,0,countBing);
		Arrays.sort(numJiao,0,countJiao);
		Collections.sort(list);
		for(int i=0;i<countJiao;i++)
		{
			System.out.print(numJiao[i]+" ");
		}
		System.out.println();
		for(int i=0;i<countBing;i++)
		{
			System.out.print(numBing[i]+" ");
		}
		System.out.println();
		Integer []num2 = list.toArray(new Integer[list.size()]);
		for(int element: num2)
		{
			System.out.print(element+" ");
		}
		System.out.println();
		
	}

}
发布了4 篇原创文章 · 获赞 0 · 访问量 28

猜你喜欢

转载自blog.csdn.net/weixin_43824233/article/details/105603234