1065. 单身狗(25)

题目链接：https://www.patest.cn/contests/pat-b-practise/1065
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。

输入格式：

输入第一行给出一个正整数N（<=50000），是已知夫妻/伴侣的对数；随后N行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（<=10000），为参加派对的总人数；随后一行给出这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 <cstdio>
using namespace std;
int cp[100001]={-1};//下标为每个人的ID存储已配对的人的配偶信息，元素值为配偶的ID
int flag[100001]={0};//每个人是否参加party
int ID[10000];//保存到场人的ID
int main(int argc, char const *argv[])
{
    int n,m;
    int p1,p2;
    cin>>n;
    for(int i=0;i<n;i++){
        scanf("%05d %05d",&p1,&p2);
        cp[p1]=p2,cp[p2]=p1;//配对
    }
    cin>>m;
    int tmp;
    int count=0;
    for(int i=0;i<m;i++){
        scanf("%05d",&tmp);//输入到场人的ID
        flag[tmp]=1;//该人已到场
        ID[i]=tmp;
    }
    for(int i=0;i<m;i++){//统计单身狗的个数
        //单身狗即为没有配偶，或者有配偶但是配偶没有参加party
        if(cp[ID[i]]==-1||(cp[ID[i]]!=-1&&flag[cp[ID[i]]]==0))
            count++;
    }
    cout<<count<<endl;
    int is1=0;//控制末尾不能有空格
    for(int i=0;i<100001;i++){//输出每个单身狗
        if(flag[i]==1&&(cp[i]==-1||(cp[i]!=-1&&flag[cp[i]]==0))){
            if(is1==0){
                printf("%05d",i);
                is1=1;
            }
            else
                printf(" %05d",i);
        }
    }
    return 0;
}

结果：