PAT乙级刷题之路1065 单身狗 (25分)

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

分析：

对于这种成对的数据我们可以让他们互相作为对方的标志来判断他们是否存在，因为只要两者同时存在那么一定也会都被标记

#include <iostream> 
#include <string> 
#include <vector>
#include <set>
using namespace std;

int main()
{
	int N;
	cin>>N;
	vector<int> couple(100000,-1);//存数组下标的对象 
	int a,b;
	for(int i=0;i<N;i++){
		cin>>a>>b;
		couple[a]=b;
		couple[b]=a;
	} 
	int M;
	cin>>M;
	vector<int> guest(M),isExist(100000);//前者存客人，后者看客人是否有伴侣
	for(int i=0;i<M;i++){
		cin>>guest[i];
		if(couple[guest[i]]!=-1){
			isExist[couple[guest[i]]]=1;
		}
	} 
	set<int> pass;//存落单的人 用set可以自动排序很舒服
	for(int i=0;i<M;i++){
		if(isExist[guest[i]]!=1){
			pass.insert(guest[i]);
		}
	} 
	printf("%d\n",pass.size()); 
	/*for(int i=0;i<count-1;i++){
		printf("%05d ",pass[i]);
	}
	printf("%05d",pass[count-1]);*/
	//上面是错的set不能用数组的形式输出！！！看下面如何遍历输出set容器中的value
	for(auto it=pass.begin();it!=pass.end();it++){
		if (it != pass.begin()) printf(" ");
        printf("%05d", *it);
	} 
	return 0;
}