题目
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
思路
给的时间限制不大,而且题目数据量也很少,我就直接暴力做了,按照题目的意思,把每个来参加的人遍历一遍看看有没有ta的配偶就行了,注意点一个是配偶是无向的,所以两个方向都要,还有一个是输出的时候00001之类的不要漏了0(测试点4)。
我这个是纯暴力写法,比较简单易懂,其中能优化的地方非常多,至少可以优化成O(mlog(m))的,应该还有更好的写法,后面我想一想再更
代码
#include<bits/stdc++.h>
using namespace std;
int main() {//暴力的
int n, x, m, cnt = 0, flag;
vector<int >a(100001);
cin >> n;
for(int i = 0; i < n; i++) {
cin >> x;
cin >> a[x];
a[a[x]] = x;
}
cin >> m;
int b[m + 5], c[m + 5];
for(int i = 0; i < m; i++) cin >> b[i];
for(int i = 0; i < m; i++) {
flag = 0;
for(int j = 0; j < m; j++) {
if(b[j] == a[b[i]]) {
flag = 1;
break;
}
}
if(!flag) c[cnt++] = b[i];
}
cout << cnt << endl;
sort(c, c + cnt);
for(int i = 0; i < cnt; i++){
if(i) cout << " ";
printf("%05d", c[i]);
}
return 0;
}
优化1
优化记录了一下配偶是否到来,减少了遍历次数,不用每次都遍历
更好的方法的话是用set来做,就是查询到配偶存在之后,直接使用set容器的删除就行了,不用多一个for循环查询
#include<bits/stdc++.h>
using namespace std;
int main() {
int n, x, m, cnt = 0, flag;
vector<int >a(100001);
cin >> n;
for(int i = 0; i < n; i++) {
scanf("%d", &x);
scanf("%d", &a[x]);
a[a[x]] = x;
}
cin >> m;
int b[m + 5], c[m + 5];
for(int i = 0; i < m; i++) cin >> b[i];
for(int i = 0; i < m; i++) {
flag = 0;
for(int j = 0; j < m; j++) {
if(b[j] == a[b[i]]) {
flag = 1;
break;
}
}
if(!flag) c[cnt++] = b[i];
}
cout << cnt << endl;
sort(c, c + cnt);
for(int i = 0; i < cnt; i++) {
if(i) cout << " ";
printf("%05d", c[i]);
}
return 0;
}
优化2
set做的优化2
#include<bits/stdc++.h>
using namespace std;
int main() {
int n, x, m, flag;
vector<int >a(100001);
vector<int >c(100001, 0);
cin >> n;
for(int i = 0; i < n; i++) {
scanf("%d", &x);
scanf("%d", &a[x]);
a[a[x]] = x;
}
cin >> m;
set<int >b;
for(int i = 0; i < m; i++) {
cin >> x;
c[x] = 1;
if(c[a[x]]) b.erase(a[x]);
else b.insert(x);
}
cout << b.size() << endl;
for(auto it = b.begin(); it != b.end(); it++) {
if(it != b.begin()) cout << " ";
printf("%05d", *it);
}
return 0;
}