PTA 1065 单身狗 (25分)

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;

猜你喜欢

转载自blog.csdn.net/m0_46368082/article/details/108174457