BZOJ 2333: [SCOI2011]棘手的操作

题目描述

真的是个很棘手的操作。。

注意每删除一个点，就需要clear一次。

#include<complex>
#include<cstdio>
using namespace std;
const int N=3e5+;
int n,m,rot,add;
//add记录所有节点增加的权值
int q[N<<];
inline int qread()
{
    int x=,j=;
    char ch=getchar();
    while(ch<'' || ch>''){if(ch=='-')j=-;ch=getchar();}
    while(ch>='' && ch<=''){x=x*+ch-'';ch=getchar();}
    return x*j;
}
struct Leftist
{
    int fat[N],son[N][],key[N],dis[N],mark[N];
    void clear(int x)
    {
        fat[x]=son[x][]=son[x][]=;
    }
    void PushDown(int x)
    {
        if(!mark[x])return;
        if(son[x][])
        {
            key[son[x][]]+=mark[x];
            mark[son[x][]]+=mark[x];
        }
        if(son[x][])
        {
            key[son[x][]]+=mark[x];
            mark[son[x][]]+=mark[x];
        }
        mark[x]=;
    }
    int Merge(int a,int b)
    {
        if(!a || !b)return a+b;
        if(key[a]<key[b])swap(a,b);
        PushDown(a);
        son[a][]=Merge(son[a][],b);
        fat[son[a][]]=a;
        if(dis[son[a][]]>dis[son[a][]])swap(son[a][],son[a][]);
        dis[a]=dis[son[a][]]+;
        return a;
    }
    int Sum(int x)
    {
        int res=;
        while(x=fat[x])res+=mark[x];
        return res;
    }
    int Top(int x)
    {
        while(fat[x])x=fat[x];
        return x;
    }
    int Pop(int x)
    {
        PushDown(x);
        int q=fat[x],p=Merge(son[x][],son[x][]);
        if(q)son[q][son[q][]==x]=p;
        fat[p]=q;
        while(q)
        {
            if(dis[son[q][]]>dis[son[q][]])
                swap(son[q][],son[q][]);
            if(dis[q]==dis[son[q][]]+)return rot;
            dis[q]=dis[son[q][]]+;
            p=q;q=fat[q];
        }
        return p;
    }
    int AddPoint(int x,int v)
    {
        int t=Top(x);
        if(t==x)
            if(!son[x][] && !son[x][])
            {
                key[x]+=v;
                return x;
            }
            else
                if(son[x][])t=son[x][];
                else t=son[x][];
        Pop(x);
        key[x]+=v+Sum(x);
        clear(x);
        return Merge(Top(t),x);
    }
    int Build()
    {
        int head=,tail=;
        for(int i=;i<=n;i++)
            q[++tail]=i;
        int x,y;
        while(tail-head)
        {
            x=q[head++];y=q[head++];
            q[++tail]=Merge(x,y);
        }
        return q[head];
    }
}t1,t2;
//对于F3操作,需要把所有堆的根拿出来建一个新的堆
//新开一棵t2来维护这些点
int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++)
        t1.key[i]=t2.key[i]=qread();
    rot=t2.Build();
    char c[];int x,y,r1,r2;
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",c);
        if(c[]=='U')
        {
            x=qread();y=qread();
            r1=t1.Top(x);r2=t1.Top(y);
            if(r1!=r2)
            {
                int tmp=t1.Merge(r1,r2);
                if(tmp==r1)t2.Pop(r2);
                else t2.Pop(r1);
                /*
                合并时,若树a并到了b上,则将a从第二棵树中删除,反之则删除b
                被删去的点还是可能会成为最大值的,可它已经被删除了....
                它再次成为最大值只有可能是在A1操作,将它再插入就好了
                */
            }
        }
        else if(c[]=='A' && c[]=='')
        {
            x=qread();y=qread();
            rot=t2.Pop(t1.Top(x));
            int tmp=t1.AddPoint(x,y);
            t2.key[tmp]=t1.key[tmp];
            t2.clear(tmp);
            rot=t2.Merge(rot,tmp);
        }
        else if(c[]=='A' && c[]=='')
        {
            x=qread();y=qread();
            r1=t1.Top(x);
            rot=t2.Pop(r1);
            //整个联通块都+y,不影响堆的性质,但mark不能及时下传,所以仍要把根删去后再插入
            t1.key[r1]+=y;t1.mark[r1]+=y;
            t2.key[r1]=t1.key[r1];
            t2.clear(r1);
            rot=t2.Merge(r1,rot);
        }
        else if(c[]=='A' && c[]=='')add+=qread();
        else if(c[]=='F' && c[]=='')
        {
            x=qread();
            printf("%d\n",t1.key[x]+add+t1.Sum(x));
        }
        else if(c[]=='F' && c[]=='')
        {
            x=qread();
            printf("%d\n",t1.key[t1.Top(x)]+add);
        }
        else printf("%d\n",t2.key[rot]+add);
    }
    return ;
}