PAT1065 单身狗 (25分) 思路记录——参考大神柳婼

1065 单身狗 (25分)

 

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

输入格式：

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

 

分析：（搬运CSDN柳婼大神博客）

柳婼：设立数组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里面的所有数就是所求的人的递增排列～

我的想法：仅在输出时候不同，因为对C++ set不太熟悉，所以输出时候可以加一个bool(是否可以输出)，第二遍的游客们第一次遍历都置为可输出（true），并存在输出数组中，若是情侣，两人输出都置为false;

输出时候判断是否可输出。

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main()
{
    vector<int> v;
    int n, m, i, j, k,cnt = 0, coup[100010],output[100010];
    bool isvis[100010],isoutput[100010];
    cin >> n;
    for (i = 0; i < n; i++)
    {
        cin >> j >> k;
        coup[j] = k;
        coup[k] = j;
    }
    cin >> m;
    cnt=m;
    for (i = 0; i < m; i++)
    {
        cin >> j;
        output[i]=j;
        isoutput[j]=true;
        isvis[j]=true;
        if(isvis[coup[j]])
        {
            isoutput[j]=false;
            isoutput[coup[j]]=false;
            cnt-=2;
        }
    }
    cout<<cnt<<endl;
    sort(output,output+m);
    if(cnt!=0) printf("%05d",output[0]);
    for(i=1;i<m;i++)
    {
        if(isoutput[output[i]])
        printf(" %05d",output[i]);
    }
    system("pause");
    return 0;
}