有n个人在公司里面工作。员工从1到n编号。每一个人属于一个部门。刚开始每一个人在自己的部门负责自己的项目,这样的话公司里面就有n个部门。
然而,公司内部出现了危机,需要合并一些部门,以提高工作效率。team(person) 表示person这个人所在的部门。有以下两种合并操作:
1. 合并 team(x) 和 team(y)。 x和 y (1≤x,y≤n)是员工编号。如果team(x) 和 team(y)是同一个部门,那么就不操作。
2. 合并team(x),team(x+1),…,team(y),x 和 y (1≤x≤y≤n)是员工编号。
有一些查询操作,查询员工x 和 y (1≤x,y≤n)是否属于同一部门。
Input
单组测试数据。
第一行有两个整数n 和 q (1≤n≤200000, 1≤q≤500000)表示员工的数目和操作数目。
接下来q行,每行的格式是type x y。type∈{1,2,3}。如果type=1 或者 type=2,那么表示第一种或者第二种合并操作。如果type=3,表示查询员工x和y是否属于同一部门。
Output
对于第三种查询,如果属于同一部门输出YES,否则输出NO。
Input示例
样例输入1
8 6
3 2 5
1 2 5
3 2 5
2 4 7
2 1 2
3 1 7
Output示例
样例输出1
NO
YES
YES
题解
链表优化区间合并,每个点只会被合并一次。
代码
#include<bits/stdc++.h>
#define mod 666623333
#define inf 10000000
typedef long long ll;
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int f[200005],pre[200005],n,q;
int find(int i)
{
if (f[i]!=i) f[i]=find(f[i]);
return f[i];
}
inline void Union1(int x,int y)
{
x=find(x),y=find(y);
f[x]=y;
}
inline void Union2(int x,int y)
{
if (x>y) swap(x,y);
while (pre[y]>=x)
{
int f1=find(pre[y]),f2=find(y);
f[f1]=f2;
int t=pre[y];
pre[y]=pre[t];
y=t;
}
}
int main()
{
n=read();q=read();
for (int i=1;i<=n;i++) pre[i]=i-1,f[i]=i;
while (q--)
{
int opt=read(),x=read(),y=read();
if (opt==1) Union1(x,y);
else if (opt==2) Union2(x,y);
else
{
x=find(x),y=find(y);
if (x==y) puts("YES");else puts("NO");
}
}
return 0;
}