PAT题解——Basic Level——1065 单身狗

题目链接

https://pintia.cn/problem-sets/994805260223102976/problems/994805266942377984。

题面

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

输入

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

题目分析

解题思路

开一个足够大的数组，将读入的 M 个数据进行标记。就是数据中是情侣的标记出来，最后将没有给标记的数输出即可。

数学知识

应该没有吧。

数据范围分析

N≤ 50,000，用数组记录情侣关系，也可以用STL的vector。比如 11111 和 22222 是情侣，哪么 couple[11111] = 22222 同时 couple[22222] = 11111。

M≤ 10,000，用数组记录参加的客人，也可以用STL的vector。当读入客人数据的时候，要记得标记对应的情侣是否到场。从 M 的大小，我们可以发现，遍历客人数组的时间复杂度最高不能超过O(nlogn)，否则可能就是TLE。

标记，可以采用数组，也可以用STL的vector。

P.S. 使用vector的好处是，定义变量的时候可以直接初始化为 -1。使用数组只能初始化为 0。

坑点

坑点1

对应一个 ID 号，为 5 位数字（从 00000 到 99999），这个要特别注意。

1、标记数组的大小应该是10^6，同时该数组的初始值要设置为 -1，不能为 0。因为情侣的 ID 可能是 0。

2、输出的时候，要注意使用占位符。这个是中文理解的问题。题目既然强调了 00000 这样的格式。

坑点2

行的首尾不得有多余空格

这个是PAT的惯用手段。输出的时候注意就行，大部分比赛不会管这个。

AC代码

//1065 单身狗
//https://pintia.cn/problem-sets/994805260223102976/problems/994805266942377984
#include <cstdio>
#include <cstring>
#include <vector>
#include <set>

const int MAXN = 1e6+2;
const int MAXM = 1e4;
int couple[MAXN] = {};//保存伴侣信息。0表示没有伴侣。
bool bExist[MAXN] = {};//存在标志位。
int guest[MAXM] = {};//客人

int main() {
    int n;
    scanf("%d", &n);

    //由于couple范围从 00000 到 99999
    //std::vector<int> couple(MAXN, -1);
    memset((void *)couple, -1, MAXN);

    //保存伴侣
    int i;
    int id1, id2;
    for (i=0; i<n; i++) {
        scanf("%d %d", &id1, &id2);
        couple[id1] = id2;
        couple[id2] = id1;
    }

    int m;
    scanf("%d", &m);

    for (i=0; i<m; i++) {
        scanf("%d", &guest[i]);
        //判断是否存在伴侣
        if (-1 != couple[guest[i]]) {
            bExist[couple[guest[i]]] = true;
        }
    }

    //遍历查找
    std::set<int> ans;
    for (i=0; i<m; i++) {
        if (false == bExist[guest[i]]) {
            ans.insert(guest[i]);
        }
    }

    //输出
    printf("%d\n", ans.size());
    std::set<int>::iterator it;
    for (it=ans.begin(); it!=ans.end(); it++) {
        if (it!=ans.begin()) {
            printf(" ");
        }
        printf("%05d", *it);
    }

    return 0;
}