PAT菜鸡进化史_乙级_1065
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
思路:
恶意满满的一题哎!
不过难度不是很高,我是用了一个大数组存了下所有人的partner以及这个人来没来。注意就是因为人太多了,数组太大,需要是static变量才可以、
Code:
#include <iostream>
#include <vector>
#include <set>
#include <iomanip>
struct people{
int partner, tag;
people(){
partner = -1; // partner's id
tag = 0; // whether partner is coming
}
};
int main(){
using namespace std;
// input the couple
int n, p1, p2;
cin >> n;
static people all[1000000];
for (int i = 0; i < n; i++){
cin >> p1 >> p2;
all[p1].partner = p2;
all[p2].partner = p1;
}
// input the coming people
cin >> n;
vector<int> come(n);
for (int i = 0; i < n; i++){
cin >> come[i];
all[come[i]].tag = 1;
}
// display the "single-dog"
set<int> single;
for (int i = 0; i < n; i++){
if (all[come[i]].partner == -1 || all[all[come[i]].partner].tag == 0)
single.insert(come[i]);
}
cout << single.size() << endl;
for (set<int>::iterator it = single.begin(); it != single.end(); it++){
if (it != single.begin())
cout << ' ';
printf("%05d", *it);
}
return 0;
}