并查集的合并操作

并查集的合并操作

例题:CodeForces - 566D D - Restructuring Company
题目需要注意:并查集合并的时候并不是简单地两个点之间的合并,其中还加入了区间的操作,如果还是按照原来的合并操作,那么时间复杂度会比较高

提示:

提示:很显然是并查集,只是type==2是对应的是区间操作,但发现如果是在一个集合里了还要操作就是浪费,可以把已经在一个区间的压缩,用一个pre数组,pre[i]表示数i的前面和i不是一个集合的与i最近的数;这样可以一次合并后下一次直接跳过中间多余的;在合并时可以省去多余的时间消耗。
代码:

#include
#include
#include
#include
#include
using namespace std;
int f[200200],b[200200];
void init(int x){
	for(int i=1;i<=x;i++){
		f[i]=i;
		b[i]=i-1;//这里建立b数组就是为了记录i前面离i最近的和他父节点不同的那个节点,这样在进行区间合并操作时,会节省很多时间
	}
}
int find(int x){
	if(x==f[x])
	    return x;
	else{
		f[x]=find(f[x]);
		return f[x];
	}
}
void merge(int x,int y){
	int t1=find(x);
	int t2=find(y);
	if(t1!=t2)
	    f[t1]=t2;
}
int main(){
	int n,q;
	scanf("%d %d",&n,&q);
	init(n);
	int l,ll,lll;
	while(q--){
		scanf("%d %d %d",&l,&ll,&lll);
		if(l==3){
			if(find(ll)==find(lll))
			    printf("YES\n");
			else
			    printf("NO\n");
		}
		else if(l==1){
			merge(ll,lll);		
}//这里不需要考虑到对数组b的操作,在程序运行时即使检测到虽然两个单位的父节点相同但是b[i]还是指向它,就让他多进行一次操作就好,不会浪费太多的时间。

		else{
			int r;
			for(int i=lll;i>=ll;i=r){
				r=b[i];//进行区间合并
				if(r

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