[PAT乙级]1065. 单身狗(25)

1065. 单身狗(25)

原题链接

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。

输入格式：

输入第一行给出一个正整数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>
#include <map>
#include <vector>
using namespace std;

int main()
{
    int n;
    cin >> n;
    map<int, int> Map;//存储每对夫妻
    for(int i=0; i<n; i++){
        int a,b;
        scanf("%05d %05d", &a, &b);
        Map[a] = b;
        Map[b] = a;
    }
    int m;
    cin >> m;
    int arr[100000] = {0};
    for(int i=0; i<m; i++){
        int temp;
        scanf("%05d", &temp);
        arr[temp] = 1;//来参加派对的人为1
    }
    vector<int> res;
    for(int i=0; i<100000; i++){
        if(arr[i]!=0 && arr[Map[i]]==0){//自己来参加派对的人并且对象没来
            res.push_back(i);
        }
    }
    printf("%d\n", res.size());
    for(int i=0; i<res.size(); i++){
        if(i!=0)
            printf(" ");
        printf("%05d", res[i]);
    }
    return 0;
}