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

思路

用空间换时间，用

vector

记录id之间的互相映射。由于vector默认初始化为0，而id范围是0~99999，可将id+1后存储，这样以后判断是否存在伴侣时只要看对应值是否等于0即可。

逐个读取id，若其没有伴侣或伴侣未出现，则用

set

记录；若伴侣出现把该伴侣删去，不作记录。

坑

测试点1格式错误：若总数为0，则只打印总数，后不再打印空行即可。

代码

#include <iostream>
#include <vector>
#include <set>
#include <iomanip>
using namespace std;

int main(){
int n;
cin >> n;
vector<int> couple(100000);
for (int i=0; i<n; i++){
int a, b;
cin >> a >> b;
couple[a] = b+1;
couple[b] = a+1;
}

int m;
cin >> m;
set<int> ss;
set<int>::iterator it;
for (int i=0; i<m; i++){
int a;
cin >> a;
if (couple[a]!=0 && ss.find(couple[a]-1)!=ss.end()){
ss.erase(ss.find(couple[a]-1));
}
else {
ss.insert(a);
}
}

//set会自动排序（从小到大）
cout << ss.size() << endl;
if (ss.size()>0){
it = ss.begin();
cout << setw(5) << setfill('0') << *it++;
while (it!=ss.end()){
cout << " " << setw(5) << setfill('0') << *it++;
}
cout << endl;
}
return 0;
}

• 点赞
• 收藏
• 分享
• 文章举报

zhang35 发布了117 篇原创文章 · 获赞 7 · 访问量 2551 私信 关注