bZOJ2333 【scoi2011】棘手的操作

时间限制：2秒  内存限制：128M

【问题描述】

　　有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: 输出所有节点中，权值最大的节点的权值。

【输入格式】

　　输入的第一行是一个整数N，代表节点个数。
　　接下来一行输入N个整数，a[1], a[2], …, a[N]，代表N个节点的初始权值。
　　再下一行输入一个整数Q，代表接下来的操作数。
　　最后输入Q行，每行的格式如题目描述所示。

【输出格式】

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

【输入样例】

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

【输出样例】

-10
10
10

【数据范围】

对于30%的数据，保证 N<=100，Q<=10000
对于80%的数据，保证 N<=100000，Q<=100000
对于100%的数据，保证 N<=300000，Q<=300000
对于所有的数据，保证输入合法，并且 -1000<=v, a[1], a[2], …, a[N]<=1000

【来源】

bZOJ2333

这道题是很好的一道可并堆的题（然而我开始居然用splay+并查集代替可并堆做出来了，虽然超时了2个点）。

我们看到这道题我们如果想到了模拟，那就是可并堆了，貌似还有一种很高级的做法，这我就不懂了，反正我没想到。（本蒟蒻无能为力）

这里我讲一下每个操作的做法：
U 可并堆的合并就好。
A1 可并堆单点修改，把整个堆分成4个，它的2个儿子，它的父亲已经更上面的点，它。然后修改它的值，再把4个堆合并起来就好了，注意细节（我开始就错了，调了好久）

A2 直接再堆顶加个标记就好了，然后以后记得下传就好（A1，F1，merge时都要下传），我的下传直接写在的find里面，这样就算忘了实际上也传了。

A3 开个变量直接存，不知道有没有人会用其他方法( ^_^ )

F1 下传一遍然后直接输出它的值就好。

F2 输出它所在堆的堆顶的值。

F3 我一开始是用的set，然后居然超时了，一气之下写了个线段树来维护，线段树的每个点，如果这个点是一个堆的顶点那就是它的值，其他点都为负无穷。（A1，A2，U的时候都要维护，详见代码）

这里说起很简单，但实现时很容易错细节，这里我提供了一种很弱但好用的调试方法，我写在代码里没删，可以细心看看（蜜汁好用）。

代码如下：

#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=300005;
const int inf=10000005;

int w[maxn],pa[maxn],lc[maxn]={0},rc[maxn]={0},n,m,tot=0,vis[maxn]={0},all=0;
int maxv[maxn*2],ch[maxn*2][2],cnt=0,root;

void up(int now){
    maxv[now]=max(maxv[ch[now][1]],maxv[ch[now][0]]);
}

void out(int now,int l,int r,int i){
    if(l==r) {maxv[now]=-inf;return;}
    int m=(l+r)>>1;
    if(i<=m) out(ch[now][0],l,m,i);
    else out(ch[now][1],m+1,r,i);
    up(now);
}

void in(int now,int l,int r,int i,int x){
    if(l==r) {maxv[now]=x;return;}
    int m=(l+r)>>1;
    if(i<=m) in(ch[now][0],l,m,i,x);
    else in(ch[now][1],m+1,r,i,x);
    up(now);
}

void buid(int &now,int l,int r){
    now=++cnt;
    maxv[now]=-inf;
    if(l==r) return;
    int m=(l+r)>>1;
    buid(ch[now][0],l,m);
    buid(ch[now][1],m+1,r);
}

void down(int x){
    if(vis[x])
    {
        int k=vis[x];vis[x]=0;
        if(lc[x])
        {
            w[lc[x]]+=k;
            vis[lc[x]]+=k;
        }
        if(rc[x])
        {
            w[rc[x]]+=k;
            vis[rc[x]]+=k;
        }
    }
}

int find(int x){
    int t=x;
    if(pa[x]) t=find(pa[x]);
    down(x);
    return t;
}

int merge(int x,int y){
    pa[x]=pa[y]=0;
    if(x==y) return x;
    if(!x||!y) return x+y;
    if(w[x]<w[y]) swap(x,y);
    down(x);
    rc[x]=merge(rc[x],y);
    swap(lc[x],rc[x]);
    if(lc[x]) pa[lc[x]]=x;
    if(rc[x]) pa[rc[x]]=x;
    return x;
}

int read(){
    int x=0,ok=0;
    char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    while((ch>='0'&&ch<='9')||ch=='-')
    {
        if(ch=='-') ok=1;
        else x=x*10+ch-'0';
        ch=getchar();
    }
    return ok?-x:x;
}

void init(){
    n=read();
    buid(root,1,n);
    for(int i=1;i<=n;i++) 
    {
        w[i]=read();
        in(root,1,n,i,w[i]);
        pa[i]=0;
    }
}

char s[5];
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    init();
    int x,y,px,py;
    m=read();
    int num=0;
    for(int i=1;i<=m;i++)
    {
        if(i==44)
        {
            int p;
            p=1;
        }
        scanf("%s",s);
        if(s[0]=='U')
        {
            x=read();y=read();
            x=find(x);y=find(y);
            if(x==y) continue;
            out(root,1,n,x);
            out(root,1,n,y);
            x=merge(x,y);
            in(root,1,n,x,w[x]);
        }
        if(s[0]=='A')
        {
            if(s[1]=='1')
            {
                x=read();y=read();
                px=pa[x];int p=find(x);
                out(root,1,n,p);
                if(lc[px]==x) lc[px]=0;
                else rc[px]=0;
                int t1=lc[x],t2=rc[x];
                pa[x]=pa[lc[x]]=pa[rc[x]]=0;
                lc[x]=rc[x]=0;

                w[x]+=y;
                px=find(px);
                t1=merge(t1,t2);x=merge(x,t1);
                px=merge(px,x);
                in(root,1,n,px,w[px]);
            }
            if(s[1]=='2')
            {
                x=read();y=read();
                x=find(x);
                out(root,1,n,x);
                w[x]+=y;
                vis[x]+=y;
                in(root,1,n,x,w[x]);
            }
            if(s[1]=='3') all+=read();
        }
        if(s[0]=='F')
        {
            num++;
            if(num==2313)
            {
                int p=1;
                p++;
            }
            if(s[1]=='1')
            {
                x=read();find(x);
                printf("%d\n",all+w[x]);
            }
            if(s[1]=='2')
            {
                x=read();
                int ans=all+w[find(x)];
                printf("%d\n",ans);
            }
            if(s[1]=='3') printf("%d\n",maxv[root]+all);
        }
        //printf("%d\n",i);
    }
    return 0;
}