51NOD 1525 重组公司 并查集

1525 重组公司
题目来源： CodeForces
基准时间限制：1 秒 空间限制：131072 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)是员工编号。

有一些查询操作，查询员工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

问题主要在于type2时 每次合并最坏情况下可能为O(n)
用一个并查集par记录i员工属于部门par[i]
用并查集head记录 head[i]到i 的员工都是同一个部门的
每次type2时
令y=find(y)
则y-1需要和y合并
直到y<=x
ps:需要大量输入输出 用加速挂+输出挂有奇效

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>

const int MAXN=200000;
int par[MAXN+5];
int head[MAXN+5];

inline int read(){
    char tmp;
    int x=0;
    while(isdigit(tmp=getchar()))
        x=x*10+tmp-'0';
    return x;
}

inline int find(int*par,int x){
    return x==par[x]?x:par[x]=find(par,par[x]);
}

inline void merge(int*par,int x,int y){
    par[find(par,y)]=find(par,x);
}

inline void mergeXY(int x,int y){
    y=find(head,y);
    while(y>x){
        int t=find(head,y-1);
        merge(head,t,y);
        merge(par,t,y);
        y=t;
    }
}

int main()
{
    //freopen("/home/lu/文档/r.txt","r",stdin);
    //freopen("/home/lu/文档/w.txt","w",stdout);
    int n=read(),q=read();
    for(int i=1;i<=n;++i){
        par[i]=head[i]=i;
    }
    int t,x,y;
    while(q--){
        t=read(),x=read(),y=read();
        if(t==1)
            merge(par,x,y);
        else
            if(t==2)
                mergeXY(x,y);
            else//printf改puts 时间马上减半啊.....
                puts(find(par,x)==find(par,y)?"YES":"NO");

    }
    return 0;
}