PAT 乙级 1065 单身狗 题解

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，说明该客人无伴侣或伴侣没有出席，计数加1。最后按升序排列输出一维数组。

 

代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

int main()
{
	int i, n, a, b, count = 0, cp[100000][2] = { 0 };
	vector<int> v;
	scanf("%d", &n);
	for (i = 0; i < n; i++)
	{
		scanf("%d %d", &a, &b);
		cp[a][0] = b;
		cp[b][0] = a;
	}
	cin >> n;
	for (i = 0; i < n; i++)
	{
		scanf("%d", &a);
		cp[a][1] = 1;
		v.push_back(a);
	}
	for (i = 0; i < n; i++)
	{
		if (!cp[cp[v[i]][0]][1]) count++;
		else v[i] = -1;
	}
	sort(v.begin(), v.end());
	printf("%d\n", count);
	for (i = n - count; i < n; i++)
	{
		if (i != n - count) printf(" ");
		printf("%05d", v[i]);
	}
	return 0;
}