试题 算法训练 集合运算

资源限制
时间限制：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.Arrays;
import java.util.Scanner;

public class Main {
public static void main(String[] args) {
  Scanner scan=new Scanner(System.in);
  int n=scan.nextInt();
  int a=0,b=0;
  int A[]=new int[n];
  for (int i = 0; i < A.length; i++) {
	A[i]=scan.nextInt();
  }
  int m=scan.nextInt();
  int B[]=new int[m];
  int Jiao[]=new int[m+n];    //交集数组
  int Bin[]=new int[m+n];     //并集数组
  int Yu[]=new int[m+n];      //B在A中余集
  int j=0,bi=0,y=0;           //三个数组指针
  for (int i = 0; i < B.length; i++) {
	B[i]=scan.nextInt();
  }
  Arrays.sort(A);  //排序
  Arrays.sort(B);

  while(a!=A.length && b!=B.length) {  //有一方到头结束
	  if(A[a]==B[b]) {
		 Bin[bi++]=B[b];
		 Jiao[j++]=B[b];
		 a++;
		 b++;
	  }else if(A[a]<B[b]) {
		 Bin[bi++]=A[a];
		 Yu[y++]=A[a++];
	  }else {
		 Bin[bi++]=B[b++]; 
	  }
  }
  if(A[A.length-1]>B[B.length-1]) {  //比较A,B尾巴谁大，说明谁还没到头
	  while(a!=A.length) {
		Bin[bi++]=A[a];
		Yu[y++]=A[a++];
	  }
  }else {
	  while(b!=B.length) {
		 Bin[bi++]=B[b++]; 
	  }
  }
  for (int i = 0; i < j; i++) {           //依次输出答案
	System.out.print(Jiao[i]+" ");
 }
 System.out.println();  
 for (int i = 0; i < bi; i++) {
		System.out.print(Bin[i]+" ");
	 }
 System.out.println();  
 for (int i = 0; i < y; i++) {
		System.out.print(Yu[i]+" ");
	 }
}
}