PAT_Basic_1065 单身狗

2022-02-07

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

输入格式：

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

输出样例：

1 2	5 10000 23333 44444 55555 88888

思路：

把夫妻/伴侣存入map中，把参加派对的人存入数组中，如果参加派对的人不在map中，将值插入set;如果参加派对的人在map中但是与她对应的人不在数组里，将值插入set。

代码：

#include<iostream>
#include<map>
#include<set>
using namespace std;

int main(){
    int n,m,t;
    map<int,int> mp;
    set<int> s;
    cin>>n;
    for(int i=1;i<=n;i++){
        int a,b;
        cin>>a>>b;
        mp.insert(make_pair(a,b));
        mp.insert(make_pair(b,a));
    }
    cin>>m;
    int b[m];
    for(int i=0;i<m;i++){
            cin>>b[i];//参加派对的id
    }
    for(int i=0;i<m;i++){
            int flag=1;
            if(mp.find(b[i])!=mp.end()){
             t=mp[b[i]];//b[i]有对象 取出对象t
             flag=0;
        }
        if(flag)//没有对象
            s.insert(b[i]);
        else{
            //有对象 看对象t来参加派对没
        for(int j=0;j<m;j++){
            if(b[j]==t){
                flag=1;//对象也一起来参加派对了 不加入s
                break;
                }
                }
        if(!flag)
        s.insert(b[i]);
        }
    }
    cout<<s.size()<<endl;
    for(auto it=s.begin();it!=s.end();it++){
        if(it!=s.begin()){
            printf(" ");
        }       
        printf("%05d",*it);
    }
    return 0;
}