集合运算

[蓝桥杯][算法训练VIP]集合运算

时间限制: 1Sec 内存限制: 128MB 提交: 108 解决: 40

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

#include <bits/stdc++.h>
#include <stdio.h>
using namespace std;
int a[1005];
int b[1005];
int c[1005];
int d[3000];
int swat(int a[],int left,int right,int x)
{
	int mid;
	while(left<=right){
		mid=(left+right)/2;
		if(a[mid]==x) return mid;
		else if(a[mid]>x) right=mid-1;
		else left=mid+1;
	}
	return -1;
}
int main()
{
	int n,m;
	cin>>n;
	for(int i=0;i<n;i++) cin>>a[i],d[i]=a[i];
	sort(a,a+n);
	cin>>m;
	for(int i=0;i<m;i++) cin>>b[i],d[i+n]=b[i];
	sort(b,b+m);
	sort(d,d+n+m);
	int k=0;
	for(int i = 0;i < n; i++){
		int t=swat(b,0,m-1,a[i]);
		if(t!=-1){
			c[k++]=a[i];
		}
	}
	for(int i=0;i<k;i++)
		cout<<c[i]<<' ';
	cout<<endl;
	int t=unique(d,d+n+m)-d;
	for(int i=0;i<t;i++)
		cout<<d[i]<<' ';
	cout<<endl;
	for(int i=0;i<n;i++){
		int p=swat(c,0,k-1,a[i]);
		if(p==-1) cout<<a[i]<<' ';
	}
	cout<<endl;
	return 0;
}