题目链接:https://pintia.cn/problem-sets/994805260223102976/problems/994805266942377984
题目描述
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入
输入第一行给出一个正整数 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
代码
#include <iostream>
#include <vector>
#include <set>
using namespace std;
int main() {
int n, a, b, m;
scanf("%d", &n);
vector<int> couple(100000, -1);
for (int i = 0; i < n; i++) {
scanf("%d%d", &a, &b);
couple[a] = b; //相互记录其伴侣的ID
couple[b] = a;
}
scanf("%d", &m);
vector<int> guest(m), isExist(100000); //isExist() 标记本人是否到场
for (int i = 0; i < m; i++) {
scanf("%d", &guest[i]);
if (couple[guest[i]] != -1) //若不是单身
isExist[guest[i]] = 1; //标记本人到场
}
set<int> s;
for (int i = 0; i < m; i++) {
if (!isExist[couple[guest[i]]]) //伴侣未到的(即没有伴侣的或有伴侣但未到的),插入s
s.insert(guest[i]);
}
printf("%d\n", s.size());
for (set<int>::iterator it = s.begin(); it != s.end(); it++) {
if (it != s.begin()) printf(" ");
printf("%05d", *it);
}
return 0;
}
【注】:代码参考柳神,更改了标记对象,现在更容易理解;