pat 乙级 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

本题有坑点，第一是输入数据可能出现00000，初始化需谨慎，第二是没有单身狗的时候不应该输出任何东西。。。。。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int couple[100010];
int guest[10010],n,temp[10010];
int check(int a,int b)
{
    for(int i=0;i<n;i++)
    {
        if(b==i)
            continue;
        if(guest[i]==couple[a])
            return 1;
    }
    return 0;
}
int main()
{
    memset(couple,-1,sizeof(couple));
    memset(guest,0,sizeof(guest));
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int t1,t2;
        cin>>t1>>t2;
        couple[t1]=t2;
        couple[t2]=t1;
    }
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>guest[i];
    }
    int cot=0;
    for(int i=0;i<n;i++)
    {
        if(!check(guest[i],i))
        {
            temp[cot++]=guest[i];
        }
    }
    sort(temp,temp+cot);
    cout<<cot<<endl;
    if(cot==0)
        return 0;
    for(int i=0;i<cot-1;i++)
    {
        printf("%05d ",temp[i]);
    }
    printf("%05d",temp[cot-1]);
    return 0;
}

---

提交代码