PAT Basic 1065. 单身狗(25)（C语言实现）

我的PAT系列文章更新重心已移至Github，欢迎来看PAT题解的小伙伴请到Github Pages浏览最新内容。此处文章目前已更新至与Github Pages同步。欢迎star我的repo。

题目

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

输入格式：

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

思路

很有......单身狗歧视的一道题! ! ! (为什么分值最多的题要提及这么沉重的话题 (；′⌒`))

设计了一个方法, 只用一个int[100000]数组来解决问题：

1. 一对夫妇ID保存在数组中两个元素中, 形成双向的映射, 这个基本所有人都是这么做的
2. 逐一读取来客ID:
   1. 如果ID有对应值, 说明有配偶, 数组相应位置标记"已签到"(使用宏定义定义为一个负值)
   2. 如果ID没有对应值, 说明是单身狗, 数组相应位置标记"落单"(另一个宏定义)
3. 遍历数组, 不管值为"已签到"和"落单"的. 如果一个ID对应有非负值(即合法ID值,说明他/她是有配偶的), 进一步如果这个值对应值是"已签到"，说明ID没有来, 而他/她的配偶来了(对照上面操作不难理解). ID这个人算是"落单",标记上
4. 再遍历一次, 值为"落单"的输出即可

看了一下网上其他人的做法, 很多都用了高级语言的功能, 如Python的dict, C++的set和map(只用C的表示完全不知道用C++的在做什么╮(╯▽╰)╭)。当然还有时间复杂度O(N*M)的.
我的方法没有使用任何的高级数据结构/算法, 写出的代码也挺简洁的, 只需要遍历两边数组, 运行时间不到10ms, 还有点小骄傲呢_

代码

最新代码@github，欢迎交流

#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;

    /* Read 'couple-list', every pair of 'index' and 'value' are a couple. */
    scanf("%d", &N);
    for(int i = 0; i < N; i++)
    {
        scanf("%d %d", &ID1, &ID2);
        couple[ID1] = ID2;
        couple[ID2] = ID1;
    }

    scanf("%d", &M);
    for(int i = 0; i < M; i++)          /* Read guest list. */
    {
        scanf("%d", &ID1);
        if(couple[ID1] >= 0)                /* If one has a mate */
            couple[ID1] = SIGNED;           /*   set SIGNED */
        else                                /* Else: not in the 'couple-list' */
            couple[ID1] = SINGLE, count++;  /*   set SINGLE */
    }

    /* If couple[ID] is >= 0 (but not signed) but couple[couple[ID]] == SIGNED
     * (signed in), this means 'ID' didn't come but his/her mate did.
     * So his/her mate is alone in the party, set to SINGLE. */
    for(int i = 0; i < 100000; i++)
        if(couple[i] >= 0 && couple[couple[i]] == SIGNED)
            couple[couple[i]] = SINGLE, count++;

    /* Those whose value is SINGLE is either a bachelor or came alone */
    printf("%d\n", count);
    for(int i = 0; i < 100000; i++)
        if(couple[i] == SINGLE)
            printf("%05d%c", i, --count ? ' ' : '\0');

    return 0;
}