CodeForces - 566D

CodeForces - 566D
t[i]表示与i不是一个祖先的最近的一个位置，合并[l,r]的时候就可以跳着合并

#include <iostream>
#include <cstdio>
#include <queue>
#include <algorithm>
#include <cmath>
#include <cstring>
#define inf 2147483647
#define N 1000010
#define p(a) putchar(a)
#define For(i,a,b) for(int i=a;i<=b;++i)

using namespace std;
int n,q,flag,x,y,xx,yy;
int d[N],t[N];
void in(int &x){
    int y=1;char c=getchar();x=0;
    while(c<'0'||c>'9'){if(c=='-')y=-1;c=getchar();}
    while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=getchar();}
    x*=y;
}
void o(int x){
    if(x<0){p('-');x=-x;}
    if(x>9)o(x/10);
    p(x%10+'0');
}

int find(int x){
    if(d[x]==x) return x;
    return d[x]=find(d[x]);
}

void merge(int l,int r){
    for(int i=l+1;i<=r;){
        xx=find(i-1);
        yy=find(i);
        d[xx]=d[yy]=min(xx,yy);
        int temp=t[i];
        t[i]=t[y];
        i=temp;
    }
}

signed main(){
    in(n);in(q);
    For(i,1,n){
        d[i]=i;
        t[i]=i+1;
    }
    while(q--){
        in(flag);in(x);in(y);
        if(flag==1){
            xx=find(x);
            yy=find(y);
            d[xx]=d[yy]=min(xx,yy);
        }
        else
            if(flag==2){
            merge(x,y);
        }
        else{
            if(find(x)==find(y)) puts("YES");
            else puts("NO");
        }
    }
    return 0;
}