树剖自用模板

版权声明:欢迎转载,请注明出处噢,谢谢 https://blog.csdn.net/DT2131/article/details/79105377
#include <bits/stdc++.h>
using namespace std;
#define ls l,mid,rt*2
#define rs mid+1,r,rt*2+1
#define mi (l+r)/2
const int MAXN=5e4+7;
const int INF=0x3f3f3f3f;
vector <int> edge[MAXN];
vector <int> linkk[MAXN];
int n,m,q;
int a[MAXN];
int tsize[MAXN],dep[MAXN],hson[MAXN],father[MAXN];//dfs1
int order[MAXN],transpos[MAXN];
int belong[MAXN],top[MAXN],cnt;//dfs2
int tree[MAXN*4];
long long lazy[MAXN*4];
int st,en;
void ini(){
    for(int i=1;i<=n;i++){
        edge[i].clear();
        belong[i]=0;
    }
    linkk[1].clear();
    dep[0]=-1;
    tsize[0]=0;cnt=0;top[1]=1;
}
void dfs1(int now,int fa){
    father[now]=fa;
    int len=edge[now].size();
    dep[now]=dep[fa]+1;
    tsize[now]=1;
    hson[now]=0;
    for(int i=0;i<len;i++){
        int pos=edge[now][i];
        if(pos!=fa){
            dfs1(pos,now);
            tsize[now]+=tsize[pos];
            if(tsize[pos]>tsize[hson[now]]) hson[now]=pos;
        }
    }
    return ;
}
void dfs2(int now,int bel,int fa){
    belong[now]=bel;
    linkk[bel].push_back(now);
    int len=edge[now].size();
    if(hson[now]!=0)
        dfs2(hson[now],bel,now);
    for(int i=0;i<len;i++){
        int pos=edge[now][i];
        if(pos!=now&&pos!=fa&&pos!=hson[now]){
            cnt++;
            linkk[cnt].clear();
            top[cnt]=pos;
            dfs2(pos,cnt,now);
        }
    }
}
void push_down(int rt){
    if(lazy[rt]!=0){
        lazy[rt*2]+=lazy[rt];
        lazy[rt*2+1]+=lazy[rt];
        lazy[rt]=0;
    }
}
void build(int l,int r,int rt){
    lazy[rt]=0;
    if(l==r){
        tree[rt]=a[order[l]];
        return ;
    }
    int mid=mi;
    build(ls);build(rs);
}
void update(int l,int r,int rt,int v){
    if(l>en||r<st) return ;
    if(st<=l&&r<=en){
        lazy[rt]+=v;
        return ;
    }
    push_down(rt);
    int mid=mi;
    update(ls,v);update(rs,v);
}
long long query(int l,int r,int rt){
    if(l==r){
        return lazy[rt]+tree[rt];
    }
    push_down(rt);
    int mid=mi;

    return st<=mid?query(ls):query(rs);
}
void solveU(int x,int y,int v){
    while(1){
        if(top[belong[x]]==top[belong[y]]){
            if(dep[x]>dep[y]) swap(x,y);
            st=transpos[x];en=transpos[y];
            update(1,n,1,v);
            break;
        }else{
            if(dep[top[belong[x]]]>dep[top[belong[y]]]) swap(x,y);
            st=transpos[top[belong[y]]];en=transpos[y];
            update(1,n,1,v);
            y=father[top[belong[y]]];
        }
    }
}
int main()
{
    while(scanf("%d%d%d",&n,&m,&q)!=-1){int x,y;
        ini();
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=1;i<n;i++){
            scanf("%d%d",&x,&y);
            edge[x].push_back(y);
            edge[y].push_back(x);
        }
        dfs1(1,0);
        dfs2(1,++cnt,0);
        int ppp=0;
        for(int i=1;i<=cnt;i++){
            int len=linkk[i].size();
            for(int j=0;j<len;j++){
                order[++ppp]=linkk[i][j];
                transpos[linkk[i][j]]=ppp;
            }
        }
        build(1,n,1);
        while(q--){char op[5];int v;
            scanf("%s",op);
            if(op[0]=='I'){
                scanf("%d%d%d",&x,&y,&v);
                solveU(x,y,v);
            }else if(op[0]=='D'){
                scanf("%d%d%d",&x,&y,&v);
                solveU(x,y,-v);
            }else{
                scanf("%d",&x);
                st=en=transpos[x];
                printf("%lld\n",query(1,n,1));
            }
        }
    }
}

展开阅读全文

没有更多推荐了,返回首页