PTA 单身狗(c++)

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 <bits/stdc++.h>
using namespace std;
int id[100005]={-1};

int main() {
 int n;
 cin>>n;
 for(int i=0;i<n;i++){
  int a,b;
  cin>>a>>b;
  id[a]=b;
  id[b]=a;
 }
 cin>>n;
 int sum[n]={-1};
 //此处用id记录身份:
 //-1表示单身,-2表示不单身但不确定另一半是否到场,-3表示不单身且两人均到场
 for(int i=0;i<n;i++){
  int a;
  cin>>a;
  sum[i]=a; 
  if(id[a]>=0&&id[id[a]]==a){
   id[a]=-2;
  }else if(id[a]>=0&&id[id[a]]==-2){
   id[id[a]]=-3;
   id[a]=-3;
  }
 } 
 set<int> s;
 set<int>::iterator iter;
 int num=0;
 for(int i=0;i<n;i++){
  if(id[sum[i]]!=-3){
   s.insert(sum[i]);
   num++;
  }
 }
 cout<<num<<endl;
 for(iter=s.begin();iter!=s.end();++iter){
  if(iter==s.begin()){
   printf("%05d",*iter);
  }else{
   printf(" %05d",*iter);
  }
 }
}
©️2020 CSDN 皮肤主题: 创作都市 设计师:CSDN官方博客 返回首页