CodeForces - 566D 並查集區間更新優化

題目大意：給定n個人，q個詢問。初始狀態下，1~n個人獨立。接下來q行，每行3個數字type(1,2,3), a, b。如果type==1，則將a和b並在一個集合；如果type==2則將a~b個人並在一個集合；如果type==3，若a和b在一個集合則輸出YES，反之輸出NO

操作2：並查集區間合併優化。
nxt[i]：i的後面第一個不和i在同一集合的元素所在的集合(初始化爲i+1)。每次區間(a,b)更新時，不是暴力++i，而是直接跳到i = nxt[i]，合併(i-1,i);並且將該區間的nxt[i]置爲nxt[b];

#include <cstdio>
#include <algorithm>

using namespace std;
const int N = 2e5+10;
int fa[N];
int n, q;
int nxt[N];

int findf(int x)
{
    return x==fa[x]?x:fa[x]=findf(fa[x]);
}

void merge_(int x, int y)
{
    int f1 = findf(x);
    int f2 = findf(y);
    if(f1 != f2)
        fa[f2] = f1;
}

int main()
{
    while(~scanf("%d %d", &n, &q))
    {
        int t, a, b;
        for(int i = 0; i <= n; ++i) fa[i]=i, nxt[i]=i+1;
        while(q--)
        {
            scanf("%d %d %d", &t, &a, &b);
            if(t==1)
                merge_(a, b);
            else if(t==2)
            {
                if(a>b) swap(a, b);
                int tmp;
                for(int i = a+1; i <= b; i = tmp)
                {
                    merge_(i-1, i);
                    tmp = nxt[i];
                    nxt[i] = nxt[b];
                }
            }
            else
            {
                int f1 = findf(a);
                int f2 = findf(b);
                if(f1==f2) printf("YES\n");
                else printf("NO\n");
            }
        }
    }
    return 0;
}