poj 4088: 集合運算

二分查找

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;

int n,m;
int a[1000010];
int b[1000010];
int c[2000010];
int cnum = 0;

int cmp(const void * a,const void * b)
{
	return *(int *)a - *(int *)b;
}

bool ina(int l,int r,int x)
{
	if(l == r && a[l] != x)
		return false;
	int mid = (l+r)/2;
	if(a[mid] == x)
	{
		a[mid] = -1;
		return true;
	}
	else if(a[mid] > x)
		return ina(l,mid,x);
	else if(a[mid] < x)
		return ina(mid + 1,r,x);
}

int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
	qsort(a,n,sizeof(a[0]),cmp);
	
	scanf("%d",&m);
	for(int i=0;i<m;i++)
		scanf("%d",&b[i]);
	qsort(b,m,sizeof(b[0]),cmp);
	
	for(int i=0;i<m;i++)
		if(!ina(0,n-1,b[i]))
		{
			c[cnum] = b[i];
			cnum++;
		}
	
	for(int i=0;i<n;i++)
		if(a[i]>0)
		{
			c[cnum] = a[i];
			cnum++;
		}
	qsort(c,cnum,sizeof(c[0]),cmp);
	for(int i=0;i<cnum-1;i++)
		printf("%d ",c[i]);
	printf("%d\n",c[cnum-1]);
	return 0;
}