CF566D.Restructuring Company(并查集）

题目链接：https://vjudge.net/contest/376346#problem/D
题意：给出n个人，q个操作，一开始n个人各自为一个团队，输入a,b,c，
①a=1时，将b和c放在同一个团队中
②a=2时，将区间[b,c]中的所有人放到同一个团队中
③a=3时，询问b和c是否在同一个团队中，在的话输出YES，不在输出NO
解题思路：
a=1的时候直接合并
a=2的时候如果一个个遍历合并的话会超时，用一个nex[i]数组表示i点所在区间的右边界+1，然后在遍历合并的时候，每次合并完i的值的时候，直接跳到nex[i]即可，因为[i,nex[i]-1]这个区间里的人都已经在同一个团队了，i放入新的团队，与它在同一并查集的其他人也就相应到了新的团队
a=3的时候，直接看find(b)与find©是否相等即可。

#include<iostream>
#include<cstdio>
using namespace std;
int pa[210000];
int nex[210000];
int n,q;
int find(int x){
	return x==pa[x]?x:pa[x]=find(pa[x]);
}
void unite(int x,int y){
	int a=find(x);
	int b=find(y);
	if(a==b) return ;
	pa[b]=a;
}
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++){
		pa[i]=i;
		nex[i]=i+1;
	}
	for(int i=1;i<=q;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		if(a==1){
			unite(b,c);
		}
		if(a==2){
			if(b>c){
				int tmp=c;
				c=b;
				b=tmp;
			}
			int to;
			for(int j=b+1;j<=c;j=to){
				unite(b,j);
				to=nex[j];
				nex[j]=nex[c];
			}
		}
		if(a==3){
			if(find(b)==find(c))
				printf("YES\n");
			else
				printf("NO\n");
		}
	}
	return 0;
}