【PAT乙级】单身狗

正文索引 [隐藏]

题目描述:

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

输入描述:

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