1065 单身狗(25) - PAT乙级

1065 单身狗(25)

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。

输入格式:

输入第一行给出一个正整数 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

Solution:

建立两张Hash表, 一张情侣id之间的映射, 另外一张是用于判断要查找的id是否已经有它的爱人出现过。

就是说, 查找时, 抛开查找的几个人不说, 判断它是不是有情侣, 然后再看它的情侣是不是出现在这次晚会中了。
先它将id映射到情侣的id, 如果存在它的情侣, 删除它的情侣(自己本身不需要添加); 如果不存在情侣, 或者是情侣没来, 就把自己添加到单身狗表中。

最后剩下的就是单身狗表。

坑点是:
char* 作为key值不行, 不清楚什么原因, 就用string吧。

#include <iostream>
#include <map>
#include <string>
using namespace std;

int main() {
int N; cin >> N;
map<string, string> pairs;
for (int i = 0; i < N; i++) {
string id_a, id_b; cin >> id_a >> id_b; // why? char*不可以
pairs[id_a] = id_b;
pairs[id_b] = id_a;
}

int M; cin >> M;
map<string, int> mp;
for (int i = 0; i < M; i++) {
string id_tmp; cin >> id_tmp;

if (pairs.count(id_tmp)) {
auto it = mp.find(pairs[id_tmp]);
if (it != mp.end()) {
mp.erase(it);
} else {
mp[id_tmp] = 1;
}
} else {
mp[id_tmp] = 1;
}
}

cout << mp.size() << endl;
for (auto it = mp.begin(); it != mp.end(); it++) {
if (it != mp.begin()) cout << " ";
cout << it->first;
}
return 0;
}