PAT 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

先用一个couple记录一对情侣，比如couple[11111]=22222，然后用一个guest数组存储接下来的一行guest，guest[i]表示这个guest的id，如果这个guest[i]的另一半不是-1，那么就设就告诉另一半，你来了，比如guest[i] != -1, 则设 isSCoupleAttent[guest[i]] = 1，比如guest[i] = 11111，那么isCoupleAttent[22222] = 1，表示是在guest数组里面的，最后遍历一下guest，看一下guest[i]的另一半是否出席了

#include <cstdio>
#include <set>
#include <iterator>

using namespace std;

int main() {
    int n = 0;
    scanf("%d", &n);
    int *couple = new int[100000];
    int *isCoupleAttent = new int[100000];
    for (int i = 0; i < 100000; i++) {
        couple[i] = -1;
        isCoupleAttent[i] = 0;

    }
    for (int i = 0; i < n; i++) {
        int a = 0, b = 0;
        scanf("%d %d", &a, &b);
        couple[a] = b;
        couple[b] = a;

    }

    int m = 0;
    scanf("%d", &m);
    int *guest = new int[m];

    for (int i = 0; i < m; i++) {
        scanf("%d", &guest[i]);
        if (couple[guest[i]] != -1) {
            isCoupleAttent[couple[guest[i]]] = 1;
        }
    }

    set<int> s;
    for (int i = 0; i < m; i++) {
        if (!isCoupleAttent[guest[i]]) {
            s.insert(guest[i]);
        }
    }

    printf("%d\n", s.size());
    for (auto it = s.begin(); it != s.end(); it++) {
        if (it != s.begin()) {
            printf(" ");
        }
        printf("%05d", *it);
    }
    delete[] isCoupleAttent;
    delete[] guest;
    delete[] couple;

}