蓝桥杯算法训练VIP-集合运算 （C++代码）

#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;
}