bzoj 1455 罗马游戏

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=1455

还是对并查集的 rt 不太熟悉...

注意删去一个堆顶后把它的 fa (rt) 改成儿子合成的新堆顶,这样路径压缩也没有错了。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const maxn=1e6+5;
int n,m,fa[maxn],ls[maxn],rs[maxn],a[maxn],dis[maxn];
bool vis[maxn];
char c[5];
int rd()
{
    int ret=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
    return ret*f;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
int merge(int x,int y)
{
    if(!x||!y)return x+y;
    if(a[x]>a[y])swap(x,y);
    rs[x]=merge(rs[x],y);
    fa[rs[x]]=x;
    if(dis[ls[x]]<dis[rs[x]])swap(ls[x],rs[x]);
    if(rs[x])dis[x]=dis[rs[x]]+1;
    else dis[x]=0;
    return x;
}
void del(int x)
{
    fa[ls[x]]=ls[x]; fa[rs[x]]=rs[x];//!
    fa[x]=merge(ls[x],rs[x]);//改变 fa[x]! 
    ls[x]=rs[x]=dis[x]=0; vis[x]=1;
}
int main()
{
    n=rd();
    for(int i=1;i<=n;i++)a[i]=rd(),fa[i]=i;
    m=rd();
    for(int i=1,x,y;i<=m;i++)
    {
        scanf("%s",c);
        if(c[0]=='M')
        {
            x=rd(); y=rd();
            if(vis[x]||vis[y])continue;
            int u=find(x),v=find(y);
            if(u==v)continue;//!!!
            merge(u,v);
        }
        else 
        {
            x=rd();
            if(vis[x]){printf("0\n"); continue;}
            printf("%d\n",a[find(x)]),del(find(x));
        }
    }
    return 0;
}

 

posted @ 2018-09-07 16:36  Zinn  阅读(92)  评论(0编辑  收藏  举报