“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 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
代码长度限制
16 KB
时间限制
200 ms
内存限制
64 MB
解题思路: 很有意思的一道题,我原先的做法是用两个数组保存伴侣,对应下标相同则表明两人是伴侣,然后用第三个数组保存出席派对的人的ID,对于每个人,遍历前两个数组看这个ID是否出现在伴侣数组中或者这个人的伴侣是否也出席了,哼哼,果不其然,毫不意外地运行超时了。
然后翻看别人的做法,发现了很妙的处理办法,只用一个数组partner保存伴侣,例如输入伴侣ID a和b,保存时让partner[a]=b,partner[b]=a; 在后面输入出席派对的人的ID时,用一个数组ID存储,再用一个数组进行标记,然后遍历ID数组,如果flag[ID[i]]=1但flag[partner[ID[i]]]=0即代表这个人出席了派对但他的伴侣没有出席或者这个人压根没有伴侣,将这些ID记录下来,即为单身狗ID,然后使用sort升序排序就能得到最后的结果;
注意测试点3出错:ID是从00000~99999的5位数字,当ID小于10000时,需要在前面补0,所以需要用printf控制输出格式%05d。
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
int n;
scanf("%d",&n);
int partner[100000]={0}; //存储伴侣
for(int i=0;i<n;i++)
{
int a,b;
scanf("%d %d",&a,&b);
partner[a] = b;
partner[b] = a;
}
int m;
scanf("%d",&m);
int ID[100000],flag[100000]={0}; //ID数组保存参加派对的人,flag数组标记是否出席派对
for(int i=0;i<m;i++)
{
scanf("%d",&ID[i]);
flag[ID[i]] = 1;
}
int k =0;
int result[100000]={0};
for(int i=0;i<m;i++)
{
if(flag[ID[i]]==1&&flag[partner[ID[i]]]==0) //若自己出席而伴侣没有出席或者没有伴侣
result[k++] = ID[i];
}
sort(result,result+k);
printf("%d\n",k);
for(int i=0;i<k;i++)
{
if(i!=k-1)
printf("%05d ",result[i]);
else
printf("%05d",result[i]);
}
}