【PAT乙级】单身狗
- 2019 年 11 月 8 日
- 筆記
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: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("%dn",ans.size()); for(int i = 0; i < ans.size(); i++) { printf("%s%05d",(i?" ":""),ans[i]); //i!=0前置输出空格 } return 0; }