- 单身狗(25)
时间限制
300 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。
输入格式：
输入第一行给出一个正整数N（<=50000），是已知夫妻/伴侣的对数；随后N行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（<=10000），为参加派对的总人数；随后一行给出这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
思路：把已婚人士记录到一个hash表中，互为对方id；查询阶段，每个query在已婚人士表中查看，没有查到则肯定单身，放到vector容器；查到了则放置到“等候区”，将其标记值设定为1；如果他/她配偶也在等候区那么她配偶的等候标记也等于1，那么两个人的等候标记都修改为2（表示双方都在场）；随后遍历等候区，根据等候标记值是否等于1表示是否为单身狗，是则放入单身狗的vector容器。最后，单身狗的vector容器排序，再输出结果。
注意如果没有单身狗则不要输出多余换行。题目不说这点略坑。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <cmath>
#include <set>
#include <iomanip>
#include <sstream>
#include <unordered_map>
using namespace std;
int main() {
unordered_map<string, string> mp;
int n;
cin >> n;
for (int i = 0; i < n; i++) {
string a, b;
cin >> a >> b;
mp[a] = b;
mp[b] = a;
}
int k;
cin >> k;
vector<string> v_dog;
unordered_map<string, int> wait_region;
for (int i = 0; i < k; i++) {
string q_id;
cin >> q_id;
if (mp[q_id] == "") {
v_dog.push_back(q_id);
}
else {
wait_region[q_id] = 1;
if (wait_region[mp[q_id]] == 1) {
wait_region[q_id] = 2;
wait_region[mp[q_id]] = 2;
}
}
}
unordered_map<string, int>::iterator it = wait_region.begin();
for (; it != wait_region.end(); it++) {
if (it->second == 1) {
v_dog.push_back(it->first);
}
}
cout << v_dog.size() << endl;
sort(v_dog.begin(), v_dog.end());
for (int i = 0; i < v_dog.size(); i++) {
if (i > 0) {
cout << " ";
}
cout << v_dog[i];
}
if (v_dog.size() > 0) {
cout << endl;
}
return 0;
}
/*
使用了unordered_map来做，其实用数组也OK，只不过哈希表对于字符串类型id也适用。
第一个坑：结果要排序
第二个坑：如果单身狗数量为0，则不应该输出换行。题目居然不说这一点，好坑
*/