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;
}