简单并查集

这里将介绍简单并查集。
例题部分是程序自动分析。
并查集是一种通过建立父节点和子节点的关系,来确立两者是否有联系。
a和b有关系,b和c有关系,a和d有关系,那么我想查找c和d的关系,就要用到并查集。
接下来给出一道例题,程序自动分析:

Description
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设x1,x2,x3,…代表程序中出现的变量,给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个问题中的约束条件为:x1=x2,x2=x3,x3=x4,x1≠x4,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。
现在给出一些约束满足问题,请分别对它们进行判定。
Input
输入文件的第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行:
第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。
接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj。
Output
输出文件包括t行。
输出文件的第k行输出一个字符串“YES”或者“NO”(不包含引号,字母全部大写),“YES”表示输入中的第k个问题判定为可以被满足,“NO”表示不可被满足。
Sample Input
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
Sample Output
NO
YES
这里如何用到并查集的知识呢,我们可以将存在关系的建立一个集合,然后进行不存在关系的人的判断,如果判断与已知相违背,那么就是NO,如果判断到最后也是统一的,那么就是YES。
我们需要map映射来处理庞大的数据,我们同时需要一个结构体去存放没有关系的数据,最后拿他们一一验证。
大体思路是这样,先给出代码,再给出解释。

#include<iostream>
#include<cstdio>
#include<map>
using namespace std;
map<int,int> fa;
struct father
{
	int a;
	int b;
}p[1000010];
int find(int x)
{
	if(x==fa[x])
		return fa[x];
	else 
		fa[x]=find(fa[x]);
}
int main()
{
	int n1,n2;
	scanf("%d",&n1);
	for(int j=0;j<n1;j++)
	{
		fa.clear();
		int count=0;
		int a,b,re;
		scanf("%d",&n2);
		for(int i=0;i<n2;i++)
		{
			scanf("%d%d%d",&a,&b,&re);
			if(!fa[a])
				fa[a]=a;
			if(!fa[b])
				fa[b]=b;
			if(re==1)
			{
				int f1=find(a),f2=find(b);
				if(f1!=f2)fa[f1]=f2;
			}
			else 
			{
				p[count].a=a;
				p[count].b=b;
				count++ ;
			}
			
		}
		int flag=0;
		for(int i=0;i<count;i++)
		{
			if(find(p[i].a)==find(p[i].b))
			{
				flag=1;
				break;
			}
		}
		if(flag==0)
			printf("YES\n");
		else 
			printf("NO\n");
	}
}

有两部分需要解释:
第一个是查找函数

int find(int x)
{
	if(x==fa[x])
		return fa[x];
	else 
		fa[x]=find(fa[x]);
}

因为我的map——负责存储父子关系的那个,是全局变量。
所以我可以直接写下查找父节点的过程。
这个程序在建立关系和查找关系的时候作用不大一样
在建立关系的时候,调用函数之前,将两个父节点定义了一个新的变量,因为已经定义了父节点,调用函数时会直接返回fa的值,所以得到的就是父节点,然后再main函数里将两个父节点相连。
在第二个过程中,通过将判断最上的节点,也就是祖宗节点,在这里函数起到了else语句中的作用,不断向上查找,直到找到祖宗,然后将祖宗作为返回值,然后实现判断。
还有另一个需要注意的,就是a和b连接后,到底谁是祖宗,那么,我们认为,a的映射对应着b,在做并查集查找时,如果a对应这b,那么我们就可以从a得到所以b是祖宗,这个需要注意。
return code;

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值