PTA 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

解题思路

这题可以用一个数组来表示关系，首先将数组初始化为0，并且保存好CP的信息，使得couple[a]=b，couple[b]=a，使用这种形式设置二者之间的关系。然后定义一个visitors的数组，表示来参加宴会的嘉宾。存完所有的嘉宾之后，通过couple关系进行查找，如果其couple到场，则该人不是单身狗；如果该人的couple数组中对应的值是-1，表示该人并没有couple，因此该人也是单身狗。由此进行判断，并将数据存入vector中，最后进行排序输出即可，注意输出的格式是prinf("%05d")表示输出5位，并且以0填充前面的内容。

代码如下

#include <iostream>
#include <vector>
#include <algorithm>
#define MAX 100000
using namespace std;

int main() 
{
	int N;
	scanf("%d", &N);
	int couple[MAX] = { -1 };	// 将数组初始化为-1，表示均为单身狗

	for (int i = 0; i < N; i++) 
	{
		int a, b;
		scanf("%d %d", &a, &b);		// 输入二者之间的关系，表示a和b是couple
		couple[a] = b;
		couple[b] = a;
	}

	int M;
	scanf("%d", &M);
	int visitors[MAX] = { 0 };		// 初始化visitors数组
	for (int i = 0; i < M; i++) 
	{
		int t;
		scanf("%d", &t);
		visitors[t] = 1;		// 等于1表示该visitor来参加了宴会
	}
	vector<int> v;
	for (int i = 0; i < MAX; i++) 
	{
		if (visitors[i]) 
		{
			if (couple[i] == -1 || !visitors[couple[i]])	 // 如果该人不是有couple的人或者couple没来，存入vector中
			{
				v.push_back(i);
			}
		}
	}
	sort(v.begin(), v.end());	// 排序，升序排列
	printf("%d\n", v.size());
	if (v.size() > 0) 
	{
		printf("%05d", v[0]);	// 控制输出格式
		for (int i = 1; i < v.size(); i++) 
		{
			printf(" %05d", v[i]);
		}
		printf("\n");
	}
	return 0;
}

输出结果