“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
思路
1.将所有情侣用一个set存起来,一对情侣再分别做两个映射,方便查询11111-->22222,22222-->11111
2.客人用一个数组存起来,再用set存一遍方便查询,接下来就遍历客人,判断是否落单
3.set是一颗二叉树,set的find函数是非常好用的
#include<bits/stdc++.h>
using namespace std;
#define maxn 1000010
map<int,int>mp;
set<int>s,s1;
int a[maxn];
vector<int>v;
int main(){
int n;
cin>>n;
for( int i=0; i<n; i++ ){
int v1,v2;
cin>>v1>>v2;
s.insert(v1);s.insert(v2);
mp[v1]=v2;
mp[v2]=v1;
}
int k;
cin>>k;
for( int i=0; i<k; i++ ){
cin>>a[i];s1.insert(a[i]);
}
for( int i=0; i<k; i++ ){
if(s.find(a[i])!=s.end()){
if(s1.find(mp[a[i]])!=s1.end());
else v.push_back(a[i]);
}
else v.push_back(a[i]);
}
sort(v.begin(),v.end());
cout<<v.size()<<'\n';
for( int i=0; i<v.size(); i++ ){
if(i) cout<<' ';
printf("%05d",v[i]);
}
return 0;
}