[PAT乙级]1065 单身狗 (思路+精简代码)

1065 单身狗

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

输入格式：
输入第一行给出一个正整数 N（N ≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（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

思路：有200ms的限时，而编号不太大，应当牺牲空间换取时间，故使用ID作为数组下标进行记录。
想要高效地检索，可以使用三个数组：a[ ]下标为ID，用于记录配偶；b[ ]是与输入顺序相同的出场名单；c[ ]下标为ID，用于记录出场。注意，编号从00000开始，故a[ ]的初值要设为-1
此后只要为每一对夫妻互相标记ID，然后标记出场者，最后遍历出场名单，看其中有谁无配偶（a[ID] = -1）或配偶未出场（c[a[ID]] = 0），则为单身狗。

#include <stdio.h>
#include <string.h>
#include <algorithm>
int a[100000], b[10000] = { 0 }, c[100000] = { 0 };  //a[]按ID记录配偶, b[]是出场名单， c[]按ID记录出场
int main() {
	int n, m, cnt = 0, t1, t2;
	memset(a, -1, sizeof(a));
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
        scanf("%d %d", &t1, &t2);
        a[t1] = t2;
        a[t2] = t1;     //下标为ID，记录配偶
    }
	scanf("%d", &m);
	for (int i = 0; i < m; i++) {
        scanf("%d", &b[i]);     //记录到出场名单
        c[b[i]] = 1;    //下标为ID，记录出场
    }
	std::sort(b, b + m);
	for (int i = 0; i < m; i++)
		if (c[a[b[i]]] == 0 || a[b[i]] == -1) cnt++;  //对单身狗计数，标准为配偶未出场或无配偶
	printf("%d\n", cnt);
	for (int i = 0, flag = 0; i < m; i++)
		if (c[a[b[i]]] == 0 || a[b[i]] == -1)
			flag ? printf(" %05d", b[i]) : flag = printf("%05d", b[i]);
	return 0;
}