并查集 - Restructuring Company - CodeForces - 566D

并查集 - Restructuring Company - CodeForces - 566D

题意：

$给定n个点，q次询问。$

$首行输入n和q。$

$接着q行输入包括三个整数，t，x，y。$

$t是操作的类型，x和y是点的编号。$

$询问分为三种：$

$①、t=1，表示将x和y合并到一个连通块中。$

$②、t=2，表示将区间[x,y]的节点全部合并到同一个连通块中。$

$③、t=3，表示查询x和y是否再同一个连通块中。若是，输{出}^{\prime \prime }YE{S}^{\prime \prime }；否则输{出}^{\prime \prime }N{O}^{\prime \prime }。$

Examples
Input：

8 6
3 2 5
1 2 5
3 2 5
2 4 7
2 1 2
3 1 7

Output：

NO
YES
YES

数据范围：

$$\begin{gathered} 1\le n\le 200000,1\le q\le 500000 \\ Timelimit:2000ms，Memorylimit:262144kB \end{gathered}$$

---

分析：

$利用并查集，很容易快速进行①和③操作。$

$问题在于操作②。$

$由于n和q的范围较大，操作②有可能会重复合并某段区间，导致效率降低。$

$解决方案：$

$额外开一个ne数组，来记录每个节点合并后，下一次需要合并的节点编号。$

$举例：合并区间[l,r]。$

$首次合并，我们是逐个合并，同时对于i\in [l,r]，更新ne[i]=ne[r]。$

$当我们再次合并区间[l^{\prime },r^{\prime }]时，$

$①.若l^{\prime }<l\le r^{\prime }\le r，就可以只合并[l^{\prime },l]这一部分节点，因为[l,r^{\prime }]已经合并过，ne[l]=ne[r]>r^{\prime }，此时循环会终止。$

$②.若l\le l^{\prime }\le r^{\prime }\le r，因为ne[l^{\prime }]=ne[r]>r^{\prime }，循环终止，不会进行合并。$

$③.若l<l\le r<r^{\prime }，与①同理。$

$这样就能保证同一个连通块的点只会被合并一次。$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>

using namespace std;

const int N=2e5+10;

int n,m;
int p[N];
int ne[N];

int Find(int x)
{
    if(p[x]!=x) return p[x]=Find(p[x]);
    return p[x];
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) p[i]=i,ne[i]=i+1;
    
    int t,a,b;
    while(m--)
    {
        scanf("%d%d%d",&t,&a,&b);
        int pa=Find(a),pb=Find(b);
        
        if(t==1) p[pa]=pb;
        else if(t==2)
        {
            int ni;
            for(int i=a+1;i<=b;i=ni) 
            {
                p[Find(i)]=Find(i-1);
                ni=ne[i];
                ne[i]=ne[b];
            }
        }
        else 
        {
            if(pa==pb) puts("YES");
            else puts("NO");
        }
    }
    
    return 0;
}

---