PAT乙级刷题/1065 单身狗/C++实现

一、题目描述

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

输入格式：

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

二、题解代码以及提交截图

#include <iostream>
#include <map>
#include <set>
#include <iomanip>

int main()
{
    std::map<int,int> couple;
    std::set<int> member;
    std::set<int> single_member;
    int N,M;
    int array1[10000];
    std::cin >> N;
    for(int i = 0;i < N;i++){
        int couple1,couple2;
        std::cin >> couple1 >> couple2;
        couple[couple1] = couple2;
        couple[couple2] = couple1;
    }
    std::cin >> M;
    for(int j = 0;j < M;j++){
        std::cin >> array1[j];
        member.insert(array1[j]);
    }
    for(int i = 0;i < M;i++){
        int A = couple[array1[i]];
        if(member.find(A) == member.end()){
            single_member.insert(array1[i]);
        }
        else{
            continue;
        }
    }
    std::cout << single_member.size() << std::endl;
    std::set<int>::iterator it;
    for (it = single_member.begin(); it != single_member.end(); it++) {
        if (it != single_member.begin()) {
            std::cout << " ";
        }
        std::cout << std::setfill('0') << std::setw(5) << *it; //注意补位
    }
}