PAT 1065 单身狗

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

AC代码：

 

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n,a,b,m;
	cin>>n;
	vector<int> couple(100000,-1);
	for(int i=0;i<n;i++){
		cin>>a>>b;
		couple[a]=b;
		couple[b]=a;
	}
	cin>>m;
	vector<int> guest(m),isExist(100000);
	for(int i=0;i<m;i++){
		cin>>guest[i];
		if(couple[guest[i]]!=-1)//如果guest[i]有对象
		isExist[couple[guest[i]]]=1;//isExist会把对象的对象（也就是自己）标记，=1表示对象有来
	}
	set<int> s;
	for(int i=0;i<m;i++){
		if(!isExist[guest[i]])//对象不在或没有对象
		s.insert(guest[i]);
	}
		cout<<s.size()<<endl;
		set<int>::iterator it;
		for( it=s.begin();it!=s.end();it++){
			if(it!=s.begin())cout<<" ";
			printf("%05d",*it);
		}
		return 0;
}