【例题】【并查集】NKOJ2107 可爱的猴子

NKOJ2107 可爱的猴子
时间限制 : 10000 MS 空间限制 : 65536 KB

问题描述
树上有n只猴子。它们编号为 1 到n。1 号猴子用它的尾巴勾着树枝。剩下的猴子都被其他的猴子用手抓着。每只猴子的每只手可以抓住另一只猴子的尾巴。从0 时刻开始，每一秒都有一只猴子松开它的一只手。这会导致一些猴子掉到地上（它们在地上也能继续松开它们的手，猴子落地的时间很短可以不计）。 你的任务是： 写一个程序，从标准输入读入猴子间抓与被抓住的关系信息，和它们放开手的顺 序，对于每一只猴子算出它落地的时间，把结果输出到标准输出。

输入格式
第一行有两个正整数n和m。n是猴子的数量，m是我们观察猴子的时间（单位为秒）。
接下来n行是初始情 况的描述。第k+1 行有两个整数表示第k个猴子抓住的猴子的编号，前一个数 代表左手抓的猴子的编号，后一个数是右手抓的猴子的编号。如果读入的数为-1 则代表猴子的手是空的。
接下来m行是对猴子观察的结果。在这m行里的第i行，有两个整数。前一个是猴子的编号，后一个是它在时刻i−1 时松开的手的编 号（1-左手，2-右手）。

输出格式
输出n个整数，每行一个。第i行表示第i个猴子落地的时间，如果在观察结束前猴子没有落地，那么输出-1

样例输入
3 2
-1 3
3 -1
1 2
1 2
3 1

样例输出
-1
1
1

提示
1≤n≤200000,1≤m≤400000

思路：
离线，从后往前讨论。
带权并查集，权值为与父节点连结的最早时间。
注意合并时往1合并，且1集合中权值赋为inf。

#include<cstdio>
#include<iostream>
using namespace std;
const int needn=200003;
const int needm=400003;
const int inf=1e9;

int n,m;
int son[3][needn];
struct fy{int id,son;} pr[needm];
bool mark[3][needn];

//..............................................
inline void in_(int &d)
{
    char t=getchar();bool mark=false;
    while(t<'0'||t>'9') {if(t=='-') mark=true;t=getchar();}
    for(d=0;!(t<'0'||t>'9');t=getchar()) d=(d<<1)+(d<<3)+t-'0';
    if(mark) d=-d;
}
int be[needn];
int ans[needn];

int getbe(int x)
{
    if(be[x]==x) return x;
    int temp=getbe(be[x]);
    ans[x]=min(ans[x],ans[be[x]]);
    return be[x]=temp;
}

int fa(int u,int v,int val)
{
    int f=getbe(u),s=getbe(v);
    if(f!=s) 
    {
        if(f==1) be[s]=f,ans[s]=val;
        else be[f]=s,ans[f]=val;
    }
}
//..............................................

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) 
    {
        in_(son[1][i]),in_(son[2][i]);
        be[i]=i,ans[i]=inf;
    }
    for(int i=1;i<=m;i++)
    {
        in_(pr[i].id),in_(pr[i].son);
        mark[pr[i].son][pr[i].id]=true;
    }
    for(int i=1;i<=n;i++)
    {
        if(!mark[1][i]&&son[1][i]!=-1) fa(i,son[1][i],m);
        if(!mark[2][i]&&son[2][i]!=-1) fa(i,son[2][i],m);
    }
    for(int i=1;i<=n;i++) if(getbe(i)==1) ans[i]=inf;
    for(int i=m;i>=1;i--)
    {
        if(son[pr[i].son][pr[i].id]!=-1) fa(pr[i].id,son[pr[i].son][pr[i].id],i-1);
    }
    for(int i=1;i<=n;i++) getbe(i),printf("%d\n",(ans[i]==inf ? -1 : ans[i]));
}