题意:
1525 重组公司
1.0 秒 131,072.0 KB 80 分 5级题
有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)是员工编号。
(3)有一些查询操作,查询员工x 和 y (1≤x,y≤n)是否属于同一部门。
输入
单组测试数据。
第一行有两个整数n 和 q (1≤n≤200000, 1≤q≤500000)表示员工的数目和操作数目。
接下来q行,每行的格式是type x y。type∈{1,2,3}。如果type=1 或者 type=2,那么表示第一种或者第二种合并操作。如果type=3,表示查询员工x和y是否属于同一部门。
输出
对于第三种查询,如果属于同一部门输出YES,否则输出NO。
输入样例
样例输入1
8 6
3 2 5
1 2 5
3 2 5
2 4 7
2 1 2
3 1 7
输出样例
样例输出1
NO
YES
YES
思路:
刚开始我还以为是棵线段树,结果却是并查集,但是正解我也想到了,但是担心会不会超时,说到底还是太菜了!!!
(1)题意给出了(1)(2)(3)三种操作,我们只需要在第三种操作之后进行输出即可
(2)经典的并查集,对于每个i,设置两个属性pre[i]和par[i],pre[i]就是正常的当前i在哪个集合,par[i]就是i可以转到1~i-1的哪个位置。所以合并操作不会进行特别多!!!也就不会超时!!!
(3)具体怎么合并就看代码吧
代码实现:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5;
int n,q;
int pre[maxn];
int par[maxn];
int find(int x){
if(x == pre[x]) return x;
else return pre[x] = find(pre[x]);
}
int main(){
scanf("%d%d",&n,&q);
int op; int x,y;
for(int i = 1;i <= n;i++){
pre[i] = i; par[i] = i - 1;
}
for(int i = 1;i <= q;i++){
scanf("%d%d%d",&op,&x,&y);
if(op == 1){
int px = find(x);
int py = find(y);
if(px != py) pre[py] = px;
}
else if(op == 2){
int i = y; int now;
for(;i >= x&&(now = par[i])>= x;i = now){
pre[find(i)] = pre[find(now)];
par[i] = par[now];
}
}
else if(op == 3){
puts(find(x) == find(y)?"YES":"NO");
}
}
return 0;
}