并查集的合并操作

并查集的合并操作

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

提示：

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

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
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<ll)
				   break;
				f[find(r)]=f[find(i)];
				b[i]=b[r];//更新b[i]的值，知道跳出循环
			}
		}
	}
	return 0;
}