1065 单身狗 (PAT乙级真题)

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

输入格式：

输入第一行给出一个正整数 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的下标，另一位的编号为这个下标的值，反过来再存一次，方便之后查找使用。在随后给出的客人编号中（以编号为temp举例），他（她）的伴侣就是map[temp]，查看array[map[temp]]的值，如果结果为1，表示temp的伴侣编号之前已经给出，所以temp不是单身狗，把array[map[temp]]的值重置，并sum--; 如果结果为-1，说明temp可能是单身的，所以把array[temp]标记为1，sum++。最后遍历array，把值为1的下标数字按要求输出即可。

一开始我没有考虑array数组初始化为-1的事情，因为可能会出现00000这个编号，不过那个代码提交之后也AC了……

#include <iostream>
using namespace std;
int main() {
    int n;
    int map[100000];
    cin >> n;
    for (int i = 0; i < n; i++) {
        int temp_1, temp_2;
        cin >> temp_1 >> temp_2;
        map[temp_2] = temp_1;
        map[temp_1] = temp_2;
    }
    cin >> n;
    int array[100001] = {0};
    for(int i = 0; i < 100001; i++) {
        array[i] = -1;
    }
    int sum = 0;
    for (int i = 0; i < n; i++) {
        int temp;
        cin >> temp;
        if(array[map[temp]] == 1) {
            array[map[temp]] = -1;
            sum--;
        } else if(array[map[temp]] == -1) {
            array[temp] = 1;
            sum++;
        }
    }
    bool flag = false;
    printf("%d\n", sum);
    for(int i = 0; i < 100001; i++) {
        if(array[i] == 1) {
            if(flag) {
                cout << " ";
            }
            printf("%05d", i);
            flag = true;
        }
    }
    return 0;
}