“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式: 输入第一行给出一个正整数 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))的,应该还有更好的写法,后面我想一想再更
优化记录了一下配偶是否到来,减少了遍历次数,不用每次都遍历
更好的方法的话是用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; }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; }