PAT甲级1065. 单身狗（二分搜索+输出格式控制）

1065. 单身狗(25)

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。

输入格式：

输入第一行给出一个正整数N（<=50000），是已知夫妻/伴侣的对数；随后N行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（<=10000），为参加派对的总人数；随后一行给出这M位客人的ID，以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式：

首先第一行输出落单客人的总人数；随后第二行按ID递增顺序列出落单的客人。ID间用1个空格分隔，行的首尾不得有多余空格。

输入样例：

3
11111 22222
33333 44444
55555 66666
7
55555 44444 10000 88888 22222 11111 23333

输出样例：

5
10000 23333 44444 55555 88888

【分析】根据客人的ID，在已知的夫妻/伴侣中搜索。若找到双方，则他们均不落单；否则该客人将落单。

        数据规模限制，这里使用二分搜索效率更高；

        此外：由于ID号的范围是00000~99999，在输出时需要注意控制输出格式；如果没有客人落单，则落单客人的总人数=0，然后不再输出任何内容。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=50005;
const int maxm=10005;
int N,M;
int ans,first=1;//ans-单身狗数目 first-输出标记 
struct pair     //伴侣 
{
	int id;     //ID号 
	int pairid; //伴侣的ID号 
} p[maxn];
struct customer    //客人
{
	int id;        //ID号 
	int is_single; //是否落单 1-是 0-否 
} c[maxn];
bool cmp(struct customer c1,struct customer c2)
{
	return (c1.id<c2.id);
}
int bsearch(int k,int left,int right) //二分搜索ID号为k的客人，找不到返回-1 
{
	int mid;
	while(left<=right)
	{
		mid=(left+right)>>1;
		if(c[mid].id==k)
			return mid;
		else if(c[mid].id<k)
			left=mid+1;
		else
			right=mid-1;
	}
	return -1;
}
int main()
{
	int i;
	int idx1,idx2;
	scanf("%d",&N);
	for(i=0;i<N;i++)
		scanf("%d %d",&p[i].id,&p[i].pairid);
	scanf("%d",&M);
	for(i=0;i<M;i++)
	{
		scanf("%d",&c[i].id);
		c[i].is_single=1;
	}
	sort(c,c+M,cmp);  //按客人ID号递增排序 
	ans=M;
	for(i=0;i<N;i++)
	{
		idx1=bsearch(p[i].id,0,M-1);
		idx2=bsearch(p[i].pairid,0,M-1);
		//若同时找到伴侣双方，则他们均不落单 
		if(idx1!=-1 && idx2!=-1)
		{
			c[idx1].is_single=0;
			c[idx2].is_single=0;
			ans-=2;
		}
	}
	//落单的客人总人数 
	printf("%d\n",ans);
	//若ans不为0，递增输出他们的ID号（注意00000~99999 占5位） 
	if(ans!=0)
	{
		for(i=0;i<M;i++)
		{
			if(c[i].is_single==1)
			{
				if(first==1)
				{
					printf("%05d",c[i].id);
					first=0;
				}
				else
					printf(" %05d",c[i].id);
			}
		}
		printf("\n");
	}
	return 0;
}