PAT Basic level 1065 单身狗 (25分)

1065 单身狗 (25分)

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

输入格式：
输入第一行给出一个正整数 N（≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（≤ 10 000），为参加派对的总人数；随后一行给出这 M 位客人的 ID，以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式：
首先第一行输出落单客人的总人数；随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔，行的首尾不得有多余空格。

思路：看注释吧，单身狗无言，唯有泪千行 哈哈哈哈哈

#include <stdio.h>

#define BLANK -1
#define signed -2
#define single -3

int main()
{
    int couple[100000], count = 0, N, M, ID1, ID2;
    for(int i = 0; i < 100000; i++)
        couple[i] = BLANK;//不能设为0，因为id可能为0；
    
    scanf("%d", &N);//输入成对对数
    for(int i = 0; i < N; i++)
    {
        scanf("%d %d", &ID1, &ID2);//输入两人id
        couple[ID1] = ID2;
        couple[ID2] = ID1;//双方分别存储对方的id，形成双向映射
    }

    scanf("%d", &M);//输入参与派对总人数
    for(int i = 0; i < M; i++)         
    {
        scanf("%d", &ID1);
        if(couple[ID1] >= 0)
            couple[ID1] = signed;//如果不是单身，就标记为signed
        else
            couple[ID1] = single, count++;//是单身就标记为single
    }

    /* 如果 couple[ID] >= 0 (但是没有被标记为 signed) 但是 couple[couple[ID]]==signed
     * 说明 'ID' 没有去 但是 他的伴侣去了.
     * 所以 他的伴侣 是单独去派对的, 也应该设为 single. */
    for(int i = 0; i < 100000; i++)
        if(couple[i] >= 0 && couple[couple[i]] == signed)
            couple[couple[i]] = single, count++;

    
    printf("%d\n", count);//输出单身人数
    for(int i = 0; i < 100000; i++)
        if(couple[i] == single)
            printf("%05d%c", i, --count ? ' ' : '\0');

    return 0;
}