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

分析：

将已知的夫妻/伴侣建立map映射，分析宾客中是否出现没有伴侣的宾客并将其标记在最后使用set保存单身的宾客即可。

代码：

#include <cstdio>
#include <vector>
#include <set>

using namespace std;

int main() {
    int N, M;
    scanf("%d", &N);
    vector<int> couple(100010, -1);

    for (int i = 0; i < N; i++) {   // 建立 map 映射
        int a, b;
        scanf("%d%d", &a, &b);
        couple[a] = b;
        couple[b] = a;
    }

    scanf("%d", &M);
    vector<int> guest(M);
    vector<bool> haveCouple(100010, false);
    for (int i = 0; i < M; i++) {
        scanf("%d", &guest[i]);

        if (couple[guest[i]] != -1)    // 如果在 couple 中出现说明不是"单身狗"
            haveCouple[couple[guest[i]]] = true;
    }

    set<int> isSingle;
    for (int i = 0; i < M; i++) {
        if (!haveCouple[guest[i]]) {    // 如果是"单身狗"
            isSingle.insert(guest[i]);
        }
    }

    printf("%lu\n", isSingle.size());

    // 遍历 isSingle
    for (auto it = isSingle.begin(); it != isSingle.end(); it++) {
        if (it != isSingle.begin())
            printf(" ");
        printf("%05d", *it);
    }

    return 0;
}