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
题目分析:读取夫妻/伴侣使用map存储方便之后使用。读取查询的ID时首先使用hash数组标记出现此人,之后遍历map表,如果夫妻/伴侣两个人同时出现在hash数组中,说明这两人不是单身狗,随后标记为false。最后遍历单身狗hash数组,将值为true对应的下标即单身狗ID存储到vector中,对vector排序后输出即可。
AC代码:
#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;
const int max_n = 100010;
bool is_appear[max_n] = {false};
map<int, int> mp;
vector<int> v;
int main(){
//freopen("in.txt", "r", stdin);
int n, a, b;
scanf("%d", &n);
for(int i=0; i<n; ++i){
scanf("%d%d", &a,&b);
mp.insert(make_pair(a,b));
}
int m, tmp;
scanf("%d", &m);
for(int i=0; i<m; ++i){
scanf("%d", &tmp);
is_appear[tmp] = true;
}
for(auto it=mp.begin(); it!=mp.end(); ++it){
if(is_appear[it->first] && is_appear[it->second]){
is_appear[it->first] = false;
is_appear[it->second] = false;
}
}
for(int i=0; i<max_n; ++i){
if(is_appear[i]==true)
v.push_back(i);
}
sort(v.begin(), v.end());
printf("%d\n", v.size());
for(int i=0; i<v.size(); ++i){
printf("%05d", v[i]);
if(i!=v.size()-1)printf(" ");
}
return 0;
}