适合初步练习PAT乙级——（1065） 单身狗

适合初步练习PAT乙级——（1065） 单身狗

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。
输入格式：
输入第一行给出一个正整数 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<set>
#include<vector>
#include<algorithm>
using namespace std;
int main() {
	int n, m;
	cin >> n;
	int lover[100000] = { -1 };
	int id1, id2;

	for (int i = 0; i < n; i++) {
		cin >> id1 >> id2;
		lover[id1] = id2;
		lover[id2] = id1;
	}
	cin >> m;
	set<int>sigMan;//单生狗
	vector<int>vec;//有情侣
	for (int i = 0; i < m; i++) {//输入m位客人的ID
		int id;
		cin >> id;
		if (lover[id] == -1)//为什么不能为0，是因为题目可以从00000到99999
			sigMan.insert(id);// 参加的人不在情侣的范围，直接放入set中
		else
			vec.push_back(id);//是情侣。先放入vector中
	}
	for (int i = 0; i < vec.size(); i++) {//遍历vector,找到对应情侣，
		if (find(vec.begin(), vec.end(), lover[vec[i]]) == vec.end())
			sigMan.insert(vec[i]);//如果情侣只有一个参加，放入set中
	}
	cout << sigMan.size() << endl;
	for (set<int>::iterator it = sigMan.begin(); it != sigMan.end(); it++) {
		if (it != sigMan.begin())
			cout << " ";
		printf("%05d",*it);
	}
	return 0;
}

思路：此题主要难点是，如何在参加的人中，判断是情侣，还是情侣其中一个。
如果是情侣之一，则也是单身狗。使用set保存客人中的单身狗。如果参加的是情侣之一，先保存在vector中，然后遍历vector，查看该情侣的另一半是否参加，如果没参加，将其放入set中，输出set。