并查集

并查集:对要求合并的集合,先查找他们是否有亲戚(自己集合里面的某个元素也在其他集合里面)在其他集合,如果有那么就合并到一个更大的集合里面

//  [9/30/2013 qingezha] Union-Find 并查集
//	初始化所有的结点用整数表示,比如(0--n-1),在处理输入pair之前,他们分属于不同的组,每个结点都是孤立的
//	可以用数组表示这种关系,数组的index表示结点,而对应的值表示组号
int const NODE_NUM = 1001;
int count_set = NODE_NUM;
int arr[NODE_NUM]={0};
void UF_init()			//初始化
{ 
	for (int i=0;i<NODE_NUM;++i)
		arr[i] = i; 
} 
int find(int p)				//查找这个结点属于 哪个组
{
	if(p>=0&&p<NODE_NUM)
		return arr[p];
	else 
		return -1;
}
bool connect(int p,int q)		//判断是否连接
{
	if(p>=0 && p<NODE_NUM && q>=0 && q<NODE_NUM)
		return arr[p]==arr[q];
	else
		return false;
}
int count()					//组的个数
{
	return count_set;
}
bool union_two(int p,int q)	//合并2个,使之为一个组
{
	if(p>=0 && p<NODE_NUM && q>=0 && q<NODE_NUM)
	{
		if(arr[p]==arr[q])
			return true;
		else
			for(int i=0;i<NODE_NUM;++i)
				if(arr[i]==arr[p])
					arr[i] = arr[q];

		--count_set;
		return true;
	}
	else
		return false;

}


你可能感兴趣的:(并查集)