PAT (Basic Level) Practice （中文）1065 单身狗 (25 分)（哈希）坑：测试点1&测试点3

1065 单身狗 (25 分)

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

输入格式：

输入第一行给出一个正整数 N（≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（≤ 10 000），为参加派对的总人数；随后一行给出这 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

———————————————————————————————————————————

这题真的真的真的坑，说实话难度不大的。

测试点1：如果全部配对成功需要输出0，所以数组最开始初始为-1而不是0

测试点3：考虑有编号为0的情况，也要输出5位

先给看一下23分的wa代码（初始化为0导致的）

#include<bits/stdc++.h>
using namespace std;
int cou[100001];
int g[100001];
int sg[100001];
int main()
{
	memset(cou,0,sizeof(cou));
	memset(g,0,sizeof(g));
	memset(sg,0,sizeof(sg));	
	int pair,i,j,x,y;
	while(cin>>pair)
	{
		for(i=0;i<pair;i++)
		{
			cin>>x>>y;
			cou[x]=y;
			cou[y]=x;
		}
		
		int n;
		cin>>n;
		for(i=0;i<n;i++)
		cin>>g[i];
		
		int flag,k=0;
		for(i=0;i<n;i++)
		{
			if(cou[g[i]])
			{
				flag=0;
				for(j=0;j<n;j++)
				{
					if(cou[g[i]]==g[j])
					{
						flag=1;
						break;
					}
				}
				if(flag==0)
				{
					sg[k++]=g[i];
				}
			}
			else
			sg[k++]=g[i];
		}
		sort(sg,sg+k);
		if(k==0)
		{
			cout<<k<<endl; 
			return 0;
		}
		cout<<k<<endl;
		for(i=0;i<k;i++)
		{
			if(i==0)
			printf("%05d",sg[i]);
			else
			printf(" %05d",sg[i]);
		}
		cout<<endl;
	}
}

改正之后的ac代码+注释：

#include<bits/stdc++.h>
using namespace std;
int cou[100001];//情侣（couple） 
int g[100001];//客人 （guest） 
int sg[100001];//单身狗 （single） 
int main()
{
	memset(cou,-1,sizeof(cou));//重点！！！记得将数组初始化为-1，因为有0的情况！！！否则测试点1不通过 
	memset(g,-1,sizeof(g));
	memset(sg,-1,sizeof(sg));	
	int pair,i,j,x,y;
	while(cin>>pair)
	{
		for(i=0;i<pair;i++)//情侣之间相互标记 （哈希） 
		{
			cin>>x>>y;
			cou[x]=y;
			cou[y]=x;
		}
		
		int n;
		cin>>n;
		for(i=0;i<n;i++)//读入客人 
		cin>>g[i];
		
		int flag,k=0;
		for(i=0;i<n;i++)
		{
			if(cou[g[i]]>=0)//如果这位客人被“情侣数组”标记过
			{
				flag=0;
				for(j=0;j<n;j++)//在客人中寻找伴侣 
				{
					if(cou[g[i]]==g[j])
					{
						flag=1;
						break;
					}
				}
				if(flag==0)//虽然有伴侣但是伴侣不在客人中 
				{
					sg[k++]=g[i];
				}
			}
			else//本来就没有伴侣 
			sg[k++]=g[i];
		}
		sort(sg,sg+k);//将伴侣升序排列 
		if(k==0)//重点！！假设为0需要输出！！！ 测试点1！！！ 
		{
			cout<<k<<endl; 
			return 0;
		}
		cout<<k<<endl;
		for(i=0;i<k;i++)
		{
			if(i==0)
			printf("%05d",sg[i]);//测试点3：注意输出5位（考虑0这种特殊情况） 
			else
			printf(" %05d",sg[i]);
		}
		cout<<endl;
	}
}