1065 单身狗

1065 单身狗 - PAT (Basic Level) Practice （中文） (pintia.cn)

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

输入格式：

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

题解

1. 输入夫妻 / 伴侣信息

• 读取一个正整数 N，它代表已知夫妻 / 伴侣的对数。
• 借助 for 循环读取 N 行数据，每行包含一对夫妻 / 伴侣的 ID 号。
• 利用 map<int, int> peo 来存储夫妻 / 伴侣的对应关系，即 peo[a] = b 且 peo[b] = a，这样可以通过一个人的 ID 快速找到其伴侣的 ID。

2. 输入参加派对的客人信息

• 读取一个正整数 M，它代表参加派对的总人数。
• 运用 set<int> s 来存储参加派对的客人的 ID。set 会自动对元素进行排序，保证元素的唯一性，且按照升序排列，方便后续处理。

3. 找出落单的客人

• 遍历 set 中的每个客人的 ID。
• 对于每个 ID，检查其伴侣是否也在参加派对的客人集合中。若伴侣不在集合中，或者该客人没有伴侣（即 peo[x] 未被赋值，默认值为 0），就将其判定为落单的客人。
• 把落单的客人的 ID 存入数组 a 中，并记录落单客人的数量 count。

4. 输出结果

• 首先输出落单客人的总人数 count。
• 接着按 ID 递增顺序输出落单客人的 ID，每个 ID 采用 5 位数字的格式输出，ID 之间用一个空格分隔，行的首尾不能有多余空格。

代码

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin>>n;
    map<int,int> peo;
    for(int i=0;i<n;i++){
        int a,b;
        cin>>a>>b;
        peo[a]=b;
        peo[b]=a;
    }
    int k;
    cin>>k;
    set<int>s;
    for(int i=0;i<k;i++){
        int a;
        cin>>a;
        s.insert(a);
    }
    int a[100001];
    int count=0;
    for(auto it=s.begin();it!=s.end();it++){
        int x=*it;
        if(peo[x]==-1||s.find(peo[x])==s.end()){
            a[count++]=x;
        }
    }
    cout<<count<<endl;
    for(int i=0;i<count;i++){
        if(i)cout<<" ";
        printf("%05d",a[i]);
    }
    return 0;
}