PAT乙级1065(C++)——龙哥哥的刷题路

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

解题思路：用map存取每一对伴侣（正序和倒叙都要存取），然后利用set来存取聚会上的客人，如果map中没有这个人对应的伴侣的信息或者set中没有这个人对应的伴侣的信息，那么输出即可

#include<iostream>
#include<map>
#include<set>
#include<vector>
#include<cstdio>
using namespace std;
int main(){
	int N=0, M=0,couple1=0,couple2=0,count=0;
	set<int> cset,aset;
	map<int, int> map;
	bool flag=true;
	cin >> N;
	for (int i = 0; i < N; i++){
		cin >> couple1 >> couple2;
		map[couple1] = couple2;
		map[couple2] = couple1;
	}
	cin >> M;
	for (int i = 0; i < M; i++){
		int temp = 0;
		cin >> temp;
		cset.insert(temp);
	}
	auto iter = cset.begin();
	while (iter != cset.end()){
		int y = *iter;
		if (map.find(y) == map.end()){
			aset.insert(y);
			count++;
		}
		else{
			int temp = map[y];
			if (cset.find(temp) == cset.end()){
				aset.insert(y);
				count++;
			}
		}
		iter++;
	}
	printf("%d\n",count);
	auto aiter = aset.begin();
	while (aiter != aset.end()){
		if (!flag){
			printf(" ");
		}
		printf("%05ld",*aiter);
		aiter++;
		flag = false;
	}
	return 0;
}