1065 单身狗 (25分)
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
分析:这道题题意很好理解,主要是如何将核心算法的时间复杂度控制在O(n),刚开始用2层for循环做模拟找出“单身狗”,结果发现200ms时间限制会导致运行超时,果然模拟就会超时 。于是舍弃2层嵌套循环,新定义一个List数组来判定客人是否为单身狗成为了本题的难点所在。
AC代码
#include<iostream>
#include<vector>
#include<set>//set内部实现是红黑树,输出自动按 ID 递增顺序列出落单的客人,就免得手写快排了
using namespace std;
int main()
{
int n; cin>>n;
vector<int> couple(100005,-1);
for(int i=0;i<n;i++){
int a,b;
cin>>a>>b;
couple[a]=b;//下标表示id。
couple[b]=a;
}//couple下标输入id,就会输出其伴侣的id。
int m; cin>>m;
int a[m],list[100005]={0};
set<int> myset;
for(int i=0;i<m;i++){
cin>>a[i];//a[i]肯定是出席了的
if(couple[ a[i] ]!=-1) list[ couple[ a[i] ] ]=1;//a[i]不仅出席,并且有伴侣。list 登记伴侣的id。
} //反过来说, list内任何登记过的,有伴侣a[i],且伴侣a[i]出席。
for(int i=0;i<m;i++){
if( !list[ a[i] ] ) myset.insert(a[i]);//list[xxx]=1表示xxx有伴侣,且伴侣也出席。那么非list[xxx]就是我们要找的'单身狗'.
}
cout<<myset.size()<<endl;
for(auto it=myset.begin();it!=myset.end();it++){
if(it==myset.begin()) printf("%05d",*it);
else printf(" %05d",*it);
}
return 0;
}