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;
}