PAT Basic 1065. 单身狗

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

输入格式：

输入第一行给出一个正整数 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
   • 如果ID有对应值，说明有配偶，数组相应位置置0
   • 如果ID没有对应值，说明是单身狗，数组相应位置置-1
3. 遍历数组，不管值为0和-1的。如果一个ID对应有值，而这个值对应0，说明ID没有来，而他/她的配偶来了。也置为-1
4. 再遍历一次，值为-1的输出
5. ID是有00000的，和标记值0冲突了，解决的方法是将ID都+1（一开始没有AC的原因）

 

#include <stdio.h>
int main()
{
    int couple[100001] = {0}, N, ID1, ID2, M, count = 0;
 
    /* every pair of 'index' and 'value' are a couple.
     * record ID + 1 to avoid '00000' conflict with 0 */
    scanf("%d", &N);
    for(int i = 0; i < N; i++)
    {
        scanf("%d %d", &ID1, &ID2); ID1++; ID2++;
        couple[ID1] = ID2;
        couple[ID2] = ID1;
    }
 
    /* Record those who come. If one has a mate then set 0(which means signed in), 
     * else(means he is not even in the 'couple-list') set -1(which means bachelor) */
    scanf("%d", &M);
    for(int i = 0; i < M; i++)
    {
        scanf("%d", &ID1); ID1++;
        if(couple[ID1]) couple[ID1] = 0;
        else            {couple[ID1] = -1; count++;}
    }
 
    /* If couple[ID] is positive but couple[couple[ID]] is zero, 
     * it means 'ID' didn't come but his/her mate did(signed in). */
    for(int i = 0; i < 100001; i++) if(couple[i] > 0 && !couple[couple[i]]) 
    {
        couple[couple[i]] = -1;
        count++;
    }
 
    /* Those whose value is -1 is a bachelor or came alone */
    printf("%d\n", count);
    for(int i = 0; i < 100001; i++) if(couple[i] == -1)
        printf("%05d%c", i - 1, --count ? ' ' : '\0');
 
    return 0;
}

分析： 设立数组couple[i] = j表示i的对象是j。一开始先设置为都是-1。

设立数组isExist表示某人的对象是否来到了派对上。接收数据的时候，对于每一对a和b，将couple的a设置为b，b设置为a，表示他俩是一对。

对于每一个需要判断的人，将其存储在guest数组里面，如果它不是单身的（也就是如果它的couple[guest[i]] != -1）那么就将它对象的isExist设置为1，表示他对象的对象（也就是他自己）来到了派对。

这样所有isExist不为1的人，对象是没有来到派对的。把所有的人遍历后插入一个集合set里面，set的size就是所求的人数，set里面的所有数就是所求的人的递增排列。

#include <iostream>
#include <vector>
#include <set>
using namespace std;
int main() {
    int n, a, b, m;
    scanf("%d", &n);
    vector<int> couple(100000, -1);
    for (int i = 0; i < n; i++) {
        scanf("%d%d", &a, &b);
        couple[a] = b;
        couple[b] = a;
    }
    scanf("%d", &m);
    vector<int> guest(m), isExist(100000);
    for (int i = 0; i < m; i++) {
        scanf("%d", &guest[i]);
        if (couple[guest[i]] != -1)
            isExist[couple[guest[i]]] = 1;
    }
    set<int> s;
    for (int i = 0; i < m; i++) {
        if (!isExist[guest[i]])
            s.insert(guest[i]);
    }
    printf("%d\n", s.size());
    for (auto it = s.begin(); it != s.end(); it++) {
        if (it != s.begin()) printf(" ");
        printf("%05d", *it);
    }
    return 0;
}