bzoj2333 [SCOI2011]棘手的操作(洛谷3273)

#include<bits/stdc++.h>
#define MAXN 300005 
#define INF 0x7f7f7f7f
using namespace std;
int read(){
    int x=0,t=1;char c=getchar();
    while(c<‘0‘||c>‘9‘){if(c==‘-‘)t=-1;c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){x=x*10+c-‘0‘;c=getchar();}
    return x*t;
}
int N,Q,a[MAXN],father[MAXN],cnt1,lazy[MAXN<<2],X[MAXN],Y[MAXN],cnt2,cont,pos[MAXN],dfso[MAXN],vis[MAXN],L[MAXN],R[MAXN],last[MAXN];
char s[5];
int find(int x){return father[x]==x?x:father[x]=find(father[x]);}
struct Node{int l,r,Max;}tree[MAXN<<2];
struct Edge{int other,pre;}e[MAXN];
struct Operation{int type,x,y;}q[MAXN];
void connect(int x,int y){
    if(x==y)return;
    e[++cnt2]=(Edge){y,last[x]};
    last[x]=cnt2;
}
void DFS(int x){
    cont++;L[x]=R[x]=pos[x]=cont,dfso[cont]=x;　　　　　　　　　　　　　　//dfso是dfs序数组，pos数组是点对应的dfs序位置
    for(int i=last[x];i;i=e[i].pre)DFS(e[i].other);
}
void Pushdown(int k){
    if(!lazy[k])return;
    tree[k<<1].Max+=lazy[k],lazy[k<<1]+=lazy[k];
    tree[k<<1|1].Max+=lazy[k],lazy[k<<1|1]+=lazy[k];
    lazy[k]=0;
}
void Build_tree(int k,int l,int r){
    tree[k].Max=-INF;　　　　　　　　　　　　　　　　　　　　　　　　　　　　//初值赋成-INF
    tree[k].l=l,tree[k].r=r;    int mid=l+r>>1;
    if(l==r){tree[k].Max=a[dfso[l]];return;} 
    Build_tree(k<<1,l,mid);        Build_tree(k<<1|1,mid+1,r);
    tree[k].Max=max(tree[k<<1].Max,tree[k<<1|1].Max);
}
void Modify(int k,int l,int r,int x){
    if(tree[k].l>r||tree[k].r<l)return;
    if(l<=tree[k].l&&tree[k].r<=r){tree[k].Max+=x,lazy[k]+=x;return ;}
    Pushdown(k); 
    Modify(k<<1,l,r,x);        Modify(k<<1|1,l,r,x);
    tree[k].Max=max(tree[k<<1].Max,tree[k<<1|1].Max);
} 
int Query(int k,int l,int r){
    if(tree[k].l>r||tree[k].r<l)return -INF;　　　　　　　　　　　　　　　　//越界返回-INF
    if(l<=tree[k].l&&tree[k].r<=r)return tree[k].Max;
    Pushdown(k);
    return max( Query(k<<1,l,r),Query(k<<1|1,l,r) );
}
int main()
{
    N=read();
    for(int i=1;i<=N;i++)a[i]=read(),father[i]=i;
    Q=read();
    for(int i=1;i<=Q;i++){
        scanf("%s",s);
        if(s[0]==‘U‘){
            q[i].type=1;
            int x,y;
            q[i].x=x=read(),q[i].y=y=read();
            int fx=find(x),    fy=find(y);
            cnt1++;            father[fy]=fx;
            X[cnt1]=fx,        Y[cnt1]=fy;　　　　　　　　　　　　　　　　//存一下要连的边，是连的祖先！
        }
        if(s[0]==‘A‘){
            if(s[1]==‘1‘)q[i].type=2,q[i].x=read(),q[i].y=read();    //x点+v 
            if(s[1]==‘2‘)q[i].type=3,q[i].x=read(),q[i].y=read();    //x点联通块+v 
            if(s[1]==‘3‘)q[i].type=4,q[i].x=read();                    //全体+v 
        }
        if(s[0]==‘F‘){
            if(s[1]==‘1‘)q[i].type=5,q[i].x=read();                    //询问x点 
            if(s[1]==‘2‘)q[i].type=6,q[i].x=read();                    //询问x点所在块 
            if(s[1]==‘3‘)q[i].type=7;                                //全体询问 
        }
    }
    for(int i=cnt1;i>0;i--)connect(X[i],Y[i]); 
    for(int i=1;i<=N;i++)if(!vis[find(i)]){
        DFS(find(i));vis[find(i)]=1;　　　　　　　　　　　　　　　　　　　　　//跑DFS序
    }
    for(int i=1;i<=N;i++)father[i]=i;　　　　　　　　　　　　　　　　　　　　//复原并查集
    Build_tree(1,1,N);
    for(int i=1;i<=Q;i++){    
        if(q[i].type==1){
            int fx=find(q[i].x),fy=find(q[i].y);
            L[fx]=min(L[fx],L[fy]),        R[fx]=max(R[fx],R[fy]);　　　　//合并时维护联通块对应区间的L,R
            father[fy]=fx;
        }        
　　　　　//简单的线段树操作
        if(q[i].type==2)Modify(1,pos[q[i].x],pos[q[i].x],q[i].y);
        if(q[i].type==3){int fx=find(q[i].x);Modify(1,L[fx],R[fx],q[i].y);} 
        if(q[i].type==4)Modify(1,1,N,q[i].x);
        if(q[i].type==5)printf("%d\n",Query(1,pos[q[i].x],pos[q[i].x]));
        if(q[i].type==6){int fx=find(q[i].x);printf("%d\n",Query(1,L[fx],R[fx]));}
        if(q[i].type==7)printf("%d\n",Query(1,1,N));
    }
    return 0;
}
//haha

void Pushdown(int k){
    if(!lazy[k])return;
    tree[k<<1].Max+=k,lazy[k<<1]+=k;
    tree[k<<1|1].Max+=k,lazy[k<<1|1]+=k;
    lazy[k]=0;
}