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);
  }
 }
}