现在的位置: 首页 > 综合 > 正文

nyoj1022 合纵连横 —支持合并,删除,查找的并查集

2018年01月21日 ⁄ 综合 ⁄ 共 1689字 ⁄ 字号 评论关闭

虽然在期末考试,但是还是想A一发,来玩玩。

反正看到代码,与新的知识点,总会让自己开心,愉悦,定神。

此篇文章,不会去分析较多并查集的思想,对于其思想,大家可以参考:

讲的又好玩,又能懂。http://blog.csdn.net/dellaserss/article/details/7724401

对于并查集支持的接口。查找:代码如下 :

int find(int x){
	return x==father[x] ? x : father[x]=find(father[x]);
}	//查找时进行路径压缩。

当然也有别的写法。具体来说就是father[x]中的值代表其父亲。正如注释说的,查找时为了修改树的形状,进行了路径 压缩。

如果要支持删除的接口要修改传下来的参数为rel[x];

合并接口如下 :

int unin(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
		father[fx]=fy;
}

也不难,就是找到这个点所在集合的代表,如果 不是同一个,就进行合并。

//当然若要支持删除的操作话,合并的两个参数也要修改为rel[x],rel[y].rel[]数组为其真实的代表·;

删除接口如下:

void del(int x){
	rel[x]=idx;//rel数组储存的值是当前真的父节点编号。
	//删除时就将其父节点换成一个尚未使用的节点。(初始化时idx=n) 
	father[idx]=idx;//然后将其父节点当作这个集合的代表。 
	idx++;
}

当然删除接口还是不好想的。这里的删除还是比较有意思的,每次删除次,就让他的集合的代表rel[x]=idx修改为一个从来没有出现的值idx.

然后,我们同样修改father[idx]=idx;

完整的支持删除的接口的代码如下 .nyoj1022。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<string>
#include<cstring>
using namespace std;
const int mx=1500000;
//并查集,查找,合并,删除。 
	int father[mx],rel[mx];
int idx;
int makeSet(int n){
	for(int i=0;i<n;i++){
		father[i]=i;
		rel[i]=i;
	}
	idx=n;
}
int find(int x){
	return x==father[x] ? x : father[x]=find(father[x]);
}	//查找时进行路径压缩。 
int unin(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
		father[fx]=fy;
	//printf("%d %d %d %d\n",fx,fy,father[fx],father[fy]);
}
void del(int x){
	rel[x]=idx;//rel数组储存的值是当前真的父节点编号。
	//删除时就将其父节点换成一个尚未使用的节点。(初始化时idx=n) 
	father[idx]=idx;//然后将其父节点当作这个集合的代表。 
	idx++;
}
int main()
{
	int a,b,T=0;
	while(scanf("%d%d",&a,&b)!=EOF){
		char temp[2];
		makeSet(a);
		while(b--){
			scanf("%s",temp);
			if(temp[0] == 'U'){
				int c,d;
				scanf("%d%d",&c,&d);
				unin(rel[c],rel[d]);
			}
			else {
				int c;
				scanf("%d",&c);
				del(c); 
			}
		}
		set<int> s;
		for(int i=0;i<a;i++){
			s.insert(find(rel[i]));
		}
		printf("Case #%d: %d\n",++T,s.size());
	} 
    return 0;
}

抱歉!评论已关闭.