7-28 单身狗 (25分) 测试点1,set,find

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。

输入格式:
输入第一行给出一个正整数 N(≤50000),是已知夫妻/伴侣的对数;随后 N 行,每行给出一对夫妻/伴侣——为方便起见,每人对应一个 ID 号,为 5 位数字(从 00000 到 99999),ID 间以空格分隔;之后给出一个正整数 M(≤10000),为参加派对的总人数;随后一行给出这 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

存cp的要初始化为-1,因为有00000,这是第一个测试点
别忘了把判断条件也改成-1

#include <bits/stdc++.h>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
int cp[100000],re[10005],sum=0;
set<int> id;
int main()
{
//	freopen("D:\\LYJ.txt","r",stdin);
	int n,m,x,y;
	mem(cp,-1);//初始赋值-1,因为后面还有id00000的这种
	cin>>n;
	while(n--)
	{
		cin>>x>>y;
		cp[x]=y;
		cp[y]=x;
	}
	cin>>m;
	for(int i=0;i<m;i++)
	{
		cin>>x;
		id.insert(x);
	}
	for(auto it=id.begin();it!=id.end();it++)
	{
		if(cp[*it]==-1)
		{
			re[sum]=(*it);
			sum++;
		}
		else if(id.find(cp[*it])==id.end())
		{
			re[sum]=(*it);
			sum++;
		}
	}
	cout<<sum<<endl;
	for(int i=0;i<sum;i++)
	{
		if(0==i) printf("%05d",re[i]);
		else printf(" %05d",re[i]);
	}
	return 0;
} 

版权声明:本文为qq_44461900原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。