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
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
int handsome[100000]={
-1};
int wife[100000]={
-1};
int present[100000]={
0};
vector<int>v;
vector<int>ans;
int n;
cin>>n;
int h;
int w;
for(int i=0;i<n;i++){
cin>>h>>w;
handsome[w]=h; //伴侣互相设置
wife[h]=w;
}
int m;
cin>>m;
int num;
int cnt=0;
for(int i=0;i<m;i++){
cin>>num;
present[num]=1; //出席则设置为1
v.push_back(num);
}
for(auto it=v.begin();it!=v.end();it++){
if(present[handsome[*it]]==0&&present[wife[*it]]==0){
//他/她的妻子/丈夫没有出席
cnt++;
ans.push_back(*it);
}
}
sort(ans.begin(),ans.end());
cout<<cnt<<endl;
for(auto it=ans.begin();it!=ans.end();it++){
if (it != ans.begin()) printf(" ");
printf("%05d",*it);
}
return 0;
}
这题真是。。。测试点1和3弄了许久。
测试点1是不存在单身狗时的输出,因为又要末尾不能有空格,所以最开始这样写来输出结果:
for(auto it=ans.begin();it!=ans.end()-1;it++){
//cout<<*it<<" ";
printf("%05d ",*it);
}
//cout<<ans.back();
printf("%05d",ans.back());
测试点1就段错误了。
还是参考别人的,不要单独输出vector.back()。
测试点3,就是存在序号为00000。所以handsome和wife数组初始化不要用0,以及输出要用
printf("%05d ",*it);
或者
#include<iomanip>
cout<<setw(5)<<setfill('0')<<*it;。