【PAT乙级】单身狗

喜欢ctrl的cxk

发布于 2019-11-08 14:18:21

3870

发布于 2019-11-08 14:18:21

文章被收录于专栏：Don的成长史

本文链接：https://blog.csdn.net/weixin_42449444/article/details/89193454

题目描述：

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

输入描述：

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

解题思路：

建立一个map，其中map的key和value存放的字符串是一对夫妻/伴侣的ID。再建立一个set用来存放参加派对的人的ID，当参加派对的人没有对象，或者他的对象没有出现在set中时，就说明这是个落单客人，把所有落单的客人都放入一个叫ans的vector中。ans.size()就是落单客人的总人数，直接无脑for-each输出ans中的所有元素即可。

AC代码：

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int N;   //已知夫妻/伴侣的对数
    cin >> N;
    map<string,string> m;  //map的value存放的是key的夫妻/伴侣
    for(int i = 0; i < N; i++)
    {
        string A,B;
        cin >> A >> B;
        m[A] = B;
        m[B] = A;
    }
    int M;   //需要参加派对的人数
    cin >> M;
    set<string> s;   //set用来记录参加派对的人的ID
    for(int i = 0; i < M; i++)
    {
        string temp;
        cin >> temp;
        s.insert(temp);
    }
    vector<string> ans;   //一个存放着落单的客人ID的名单
    for(auto it : s)
    {
        if(s.count(m[it]) == 0)
        {
            ans.push_back(it);   //把落单的客人放入名单
        }
    }
    cout << ans.size() << endl;   //输出落单客人的总人数
    bool isVirgin = true;   //判断是不是第一次
    for(auto it : ans)
    {
        if(isVirgin)
        {
            cout << it;
            isVirgin = false;
        }
        else
        {
            cout << " " << it;
        }
    }
    return 0;
}

———————————————2019.07.17更新——————————————————

HBU训练营有出现这道题，我感觉我这次写的代码比之前刷PAT乙级时写的要更加简洁。

AC代码：

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int N;   //已知恋爱狗对数
    cin >> N;
    map<int,int> m;   //用来存恋爱狗信息,key是本人,value是对象
    for(int i = 0; i < N; i++)
    {
        int id1,id2;
        cin >> id1 >> id2;
        m[id1] = id2;
        m[id2] = id1;
    }
    int M;    //参加派对的总人数
    cin >> M;
    vector<int> v;   //用来存参加派对的人的信息
    for(int i = 0; i < M; i++)
    {
        int id;
        cin >> id;
        v.push_back(id);
    }
    vector<int> ans;   //落单者的信息
    for(auto it : v)
    {
        if(find(v.begin(),v.end(),m[it]) == v.end())   //落单的话
        {
            ans.push_back(it);
        }
    }
    sort(ans.begin(),ans.end());   //升序排列
    printf("%d\n",ans.size());
    for(int i = 0; i < ans.size(); i++)
    {
        printf("%s%05d",(i?" ":""),ans[i]);    //i!=0前置输出空格
    }
    return 0;
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019/04/10 ，如有侵权请联系 cloudcommunity@tencent.com 删除

key