【bzoj2333】[SCOI2011]棘手的操作 可并堆+set

Description

有N个节点，标号从1到N，这N个节点一开始相互不连通。第i个节点的初始权值为a[i]，接下来有如下一些操作：

U x y: 加一条边，连接第x个节点和第y个节点

A1 x v: 将第x个节点的权值增加v

A2 x v: 将第x个节点所在的连通块的所有节点的权值都增加v

A3 v: 将所有节点的权值都增加v

F1 x: 输出第x个节点当前的权值

F2 x: 输出第x个节点所在的连通块中，权值最大的节点的权值

F3: 输出所有节点中，权值最大的节点的权值

Input

输入的第一行是一个整数N，代表节点个数。

接下来一行输入N个整数，a[1], a[2], …, a[N]，代表N个节点的初始权值。

再下一行输入一个整数Q，代表接下来的操作数。

最后输入Q行，每行的格式如题目描述所示。

Output

对于操作F1, F2, F3，输出对应的结果，每个结果占一行。

Sample Input

3
0 0 0
8
A1 3 -20
A1 2 20
U 1 3
A2 1 10
F1 3
F2 3
A3 -10
F3

Sample Output

-10
10
10

HINT

对于30%的数据，保证 N<=100，Q<=10000

对于80%的数据，保证 N<=100000，Q<=100000

对于100%的数据，保证 N<=300000，Q<=300000

对于所有的数据，保证输入合法，并且 -1000<=v, a[1], a[2], …, a[N]<=1000

代码

#include<bits/stdc++.h>
#define ll long long
#define inf 1000000000
#define mod 65537
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;
}
const int N=300005;
int n,Q,add;
int a[N],q[N],fa[N],ls[N],rs[N],tag[N],tot;
multiset<int>s;
int find(int x)
{
    while (fa[x]) x=fa[x];
    return x;
}
void pushdown(int x)
{
    if(!tag[x])return;
    int t=tag[x],l=ls[x],r=rs[x];
    tag[x]=0;
    if(l)tag[l]+=t,a[l]+=t;
    if(r)tag[r]+=t,a[r]+=t;
}
int merge(int x,int y)
{
    if (!x) return y;
    if (!y) return x;
    if (a[x]<a[y]) swap(x,y);
    pushdown(x);
    rs[x]=merge(rs[x],y);
    fa[rs[x]]=x;
    swap(ls[x],rs[x]);
    return x;
}
void pushtag(int x)
{
    while(x)q[++tot]=x,x=fa[x]; 
    while(tot)pushdown(q[tot--]);
}
int del(int x)
{
    pushtag(x);
    int f=fa[x],t=merge(ls[x],rs[x]);
    ls[x]=rs[x]=fa[x]=0;
    if (x==ls[f]) ls[f]=t;else rs[f]=t;
    fa[t]=f;
    return find(t);
}
int main()
{
    n=read();for (int i=1;i<=n;i++) a[i]=read(),s.insert(a[i]);
    Q=read();
    while (Q--)
    {
        char ch[5];scanf("%s",ch);
        if (ch[0]=='U')
        {
            int x=find(read()),y=find(read());
            if (x!=y)
            {
                if (merge(x,y)==x)s.erase(s.find(a[y]));
                else s.erase(s.find(a[x]));
            }
        }
        else if (ch[0]=='A')
        {
            if (ch[1]=='1')
            {
                int x=read(),v=read();
                pushtag(x);s.erase(s.find(a[find(x)]));
                a[x]+=v;
                s.insert(a[merge(x,del(x))]);
            }
            else if (ch[1]=='2')
            {
                int x=read(),v=read(),f=find(x);
                tag[f]+=v;a[f]+=v;
                s.erase(s.find(a[f]-v));s.insert(a[f]);
            }
            else
            {
                int v=read();add+=v;
            }
        }
        else
        {
            if (ch[1]=='1')
            {
                int x=read();pushtag(x);printf("%d\n",a[x]+add);
            }
            else if (ch[1]=='2')
            {
                int x=read(),f=find(x);printf("%d\n",a[f]+add);
            }
            else printf("%d\n",*--s.find(inf)+add);
        }
    }
    return 0;
}