PAT --- 1065.单身狗 （25 分）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/MissXy_/article/details/88221658

1065 单身狗 （25 分）

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

输入格式：

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

思路

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

代码

#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++) {		// 将id1数字保存id2，反之
		cin >> id1 >> id2;
		lover[id1] = id2;
		lover[id2] = id1;
	}
	cin >> m;
	set<int> sigMan;
	vector<int> vec;
	for (int i = 0; i < m; i++) {		//输入参加的人
		cin >> id1;
		if (lover[id1] == -1)			//参加的人不在情侣范围内，直接放入set中
			sigMan.insert(id1);
		else
			vec.push_back(id1);			//是情侣，先放入vector中
	}
	for (int i = 0; i < vec.size(); i++) {	//遍历vector，找到对应情侣
		auto result = find(vec.begin(), vec.end(), lover[vec[i]]);
		if (result == end(vec))				//如果情侣只有一个参加，放入set中
			sigMan.insert(vec[i]);
	}
	cout << sigMan.size() << endl;
	for (auto b = sigMan.begin(); b != sigMan.end(); b++) {
		if (b != sigMan.begin())
			cout << " ";
		printf("%05d", *b);			// 一开始使用 cout<< *b; 一个测试点不过
	}
	return 0;
}