PAT1065----两个两个数据相互联系的一个解题思路

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

输入格式：

输入第一行给出一个正整数 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

/* 
 * 1. 单身狗的判断：
 *   1.1 没有对象；
 *   1.2 有对象，两人中只有一人参加派对；
 * 2. 以ID为数组下标记录情感状况；
 * 3. 有对象的人，参加派对的人做特殊标记；
 */
#include <stdio.h>
#include <stdlib.h>

int main() {
    int N, M, cp[100000];//伴侣的对数，派对总人数，记录情感状况
    scanf("%d", &N);
    for (int i = 0; i < 100000; i++) {//初始化情感状况
        cp[i] = -1; 
    }
    int a, b;
    for (int i = 0; i < N; i++) {//读取N行情侣的id 
        scanf("%d %d", &a, &b);
        cp[a] = b;//记录id对应的情侣id 
        cp[b] = a;
    }
    scanf("%d", &M);
    int all[M], cnt_dog = 0;//所有人员的id，单身狗的数量 
    for (int i = 0; i < M; i++) {//读取所有参加派对的人的id 
        scanf("%d", &all[i]);
        if (cp[all[i]] == -1) {//如果没有对象 
            cnt_dog++;
            cp[all[i]] = -3;//更新单身狗的标记 
        } else {//如果有对象，将值设为-2，表示自己参加 
            cp[all[i]] = -2; 
        } 
    }
    for (int i = 0; i < 100000; i++) {//筛选有对象的人之中的单身狗 
        if (cp[i] > -1 && cp[cp[i]] == -2) {//自己没参加，对象参加 
            cnt_dog++; 
            cp[cp[i]] = -3;//改为单身狗状态 
        }
    } 
    printf("%d
", cnt_dog);
    for (int i = 0; i < 100000 && cnt_dog > 0; i++) {
        if (cp[i] == -3) {//输出单身 
            printf("%05d", i);
            if (cnt_dog != 1) {
                printf(" ");
            }
            cnt_dog--;
        }
    }
    return 0;
}