51NOD1525重组公司(并差集)

来源:互联网 发布:电影截图拼图软件 编辑:程序博客网 时间:2024/05/17 04:12
有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∈{123}。如果type=1 或者 type=2,那么表示第一种或者第二种合并操作。如果type=3,表示查询员工x和y是否属于同一部门。Output对于第三种查询,如果属于同一部门输出YES,否则输出NO。Input示例样例输入18 63 2 51 2 53 2 52 4 72 1 23 1 7Output示例样例输出1NOYESYES
用一个pre数组记录一下前面第一个与他是不是同个部门的点。然后区间更新的时候,从y~x更新,即从后往前更新,如果该店的前一个不同的点不在(x,y)内,则结束更新,若存在则更新该点,一直重复该过程直到条件不满足为止。
#include<iostream>#include<cstring>#include<cstdlib>#include<algorithm>#include<cctype>#include<cmath>#include<ctime>#include<string>#include<stack>#include<deque>#include<queue>#include<list>#include<set>#include<map>#include<cstdio>#include<limits.h>#define MOD 1000000007#define fir first#define sec second#define fin freopen("/home/ostreambaba/文档/input.txt", "r", stdin)#define fout freopen("/home/ostreambaba/文档/output.txt", "w", stdout)#define mes(x, m) memset(x, m, sizeof(x))#define Pii pair<int, int>#define Pll pair<ll, ll>#define INF 1e9+7#define inf 0x3f3f3f3f#define Pi 4.0*atan(1.0)#define lowbit(x) (x&(-x))#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define max(a,b) a>b?a:btypedef long long ll;typedef unsigned long long ull;const double eps = 1e-9;const int maxn = 200000+5;const int maxm = 10000005;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 p[maxn],pre[maxn],N,Q,t,x,y;inline int find(int x){    return x==p[x]?x:p[x]=find(p[x]);}inline void merge(int x,int y){    x=find(x),y=find(y);    if(x!=y){        p[y]=x;    }}inline void solve(int x,int y){    int i=y;    int end;    while(i>=x&&(end=pre[i])>=x){        p[find(end)]=p[find(i)];        pre[i]=pre[end];        i=end;    }}int main(){   // fin;    N=read(),Q=read();    for(int i=1;i<=N;++i){        p[i]=i,pre[i]=i-1;    }    while(Q--)    {        t=read(),x=read(),y=read();        if(1==t){            merge(x,y);        }        else if(2==t){            solve(x,y);        }        else{           if(find(x)==find(y)){               puts("YES");           }           else{               puts("NO");           }        }    }    return 0;}
0 0
原创粉丝点击