“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
解题思路:
- 两个
map
分别是用于保存配偶映射的 couple 和出席人员 occur,检查配偶是否出现过,配偶出现过,则夫妻双方均置false。 - 向量
v
按顺序保存出席人员id
,集合s
根据occur[ v[id] ]
保存单身狗id
。
答案代码:
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <set>
using namespace std;
map<string, string> couple;
map<string, bool> occur;
int main() {
/* 1. 配偶映射 couple */
int n = 0;
cin >> n;
string male, female;
for (int i = 0; i < n; i++) {
cin >> male >> female;
couple[male] = female; couple[female] = male;
}
/* 2. 出席人员映射 occur */
int m = 0;
cin >> m;
string id;
vector<string> v;
for (int i = 0; i < m; i++) {
cin >> id;
v.push_back(id); // 出席人员向量
occur[id] = true;
if (occur[couple[id]]) { // 配偶出现过,则夫妻双方均置false
occur[id] = false; occur[couple[id]] = false;
}
}
/* 3. 单身狗集合 set */
set<string> s;
for (int i = 0; i < m; i++)
if (occur[v[i]]) s.insert(v[i]); // 按照出席顺序存储
cout << s.size() << endl;
for (auto it = s.begin(); it != s.end(); it++) {
if (it != s.begin()) cout << " ";
cout << *it;
}
return 0;
}