题目如下
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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;
}