前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Codeforces 468B

Codeforces 468B

作者头像
triplebee
发布2018-01-12 15:20:24
6020
发布2018-01-12 15:20:24
举报

很好的一道集合关系的题目,我比赛时用的是dfs,后来看到有人用并查集。

两种方法本质是一样的,但后者实现起来更方便,代码更简练。

代码语言:javascript
复制
/****************************************************
   file name: cf.cpp
   author: huangjipeng
   creat time: 2014年09月21日 星期日 22时32分55秒
***************************************************/
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ls (p<<1)
#define rs (ls|1)
#define mid ((tree[p].l+tree[p].r)>>1)
#define MAXN 1
typedef long long ll;
int n,a,b;
int num[100005];
int set[100005];
map <int,int> mp;
int find(int x)
{
	if(set[x]!=x)
		set[x]=find(set[x]);
	return set[x];
}
int main()
{
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&num[i]);
		mp[num[i]]=i;
	}
	for(int i=1;i<=(n+2);i++)
		set[i]=i;
	for(int i=1;i<=n;i++)
	{
		if(mp[a-num[i]])//如果有这个数,那么把i和这个数的下表放在一个集合中
			set[find(i)]=find(mp[a-num[i]]);
		else
			set[find(i)]=find(n+2);//否则i放到n+2集合中;
		if(mp[b-num[i]])
			set[find(i)]=find(mp[b-num[i]]);
		else
			set[find(i)]=find(n+1);//第二种情况,放到n+1集合中
	}
	if(find(n+1)==find(n+2))//如果n+1和n+2在同一集合中,说明存在元素两个集合都不属于
		cout<<"NO"<<endl;
	else
	{
		cout<<"YES"<<endl;
		for(int i=1;i<=n;i++)
		{
			cout<<(find(i)==find(n+2))<<" ";
		}
		cout<<endl;
	}
	return 0;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2014-09-22 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档