*树链剖分

今天是2017/5/22，DCDCBigBig的第九篇博文

树链剖分（+线段树）

//hdu3966
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
struct edge{
    int v,next;
}a[2000001];
int n,m,q,u,v,ww,now=0,tot=0,sum[2000001],cal[2000001],num[200001],head[200001],son[200001],siz[200001],dep[200001],fa[200001],top[200001],vis[200001],w[200001];
char ord[10];
int max(int a,int b){
    return a>b?a:b;
}
void add(int u,int v){
    a[++tot].v=v;
    a[tot].next=head[u];
    head[u]=tot;
}
void dfs1(int u,int f,int dp){
    int v;
    dep[u]=dp;
    fa[u]=f;
    siz[u]=1;
    for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
        v=a[tmp].v;
        if(v!=f){
            dfs1(v,u,dp+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[v]>siz[son[u]]){
                son[u]=v;
            }
        }
    }
}
void dfs2(int u,int tp){
    int v;
    top[u]=tp;
    vis[u]=++now;
    w[vis[u]]=u;
    if(son[u]==-1)return;
    dfs2(son[u],tp);
    for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
        v=a[tmp].v;
        if(v!=son[u]&&v!=fa[u])dfs2(v,v);
    }
}
void pushD(int rt,int len){
    if(cal[rt]){
        cal[rt<<1]+=cal[rt];
        cal[rt<<1|1]+=cal[rt];
        sum[rt<<1]+=(len-len/2)*cal[rt];
        sum[rt<<1|1]+=(len/2)*cal[rt];
        cal[rt]=0;
    }
}
void build(int l,int r,int rt){
    cal[rt]=0;
    if(l==r){
        sum[rt]=num[w[l]];
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void updata(int rt,int k,int l,int r,int L,int R){
    if(L<=l&&r<=R){
        cal[rt]+=k;
        sum[rt]+=k*(r-l+1);
        return;
    }
    int mid=(l+r)/2;
    pushD(rt,r-l+1);
    if(L<=mid)updata(rt<<1,k,l,mid,L,R);
    if(R>mid)updata(rt<<1|1,k,mid+1,r,L,R);
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
int query(int rt,int l,int r,int aa){
    if(l==r){
        return sum[rt];
    }
    int ans=0,mid=(l+r)/2;
    pushD(rt,r-l+1);
    if(aa<=mid)ans=query(rt<<1,l,mid,aa);
    else ans=query(rt<<1|1,mid+1,r,aa);
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
    return ans;
}
void update(int u,int v,int w){
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]])swap(u,v);
        updata(1,w,1,n,vis[top[u]],vis[u]);
        u=fa[top[u]];
    }
    if(dep[u]>dep[v])swap(u,v);
    updata(1,w,1,n,vis[u],vis[v]);
}
int main(){
    while(scanf("%d%d%d",&n,&m,&q)!=EOF){
        memset(head,255,sizeof(head));
        memset(son,255,sizeof(son));
        tot=now=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&num[i]);
        }
        for(int i=1;i<=m;i++){
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        dfs1(1,0,0);
        dfs2(1,1);
        build(1,n,1);
        for(int i=1;i<=q;i++){
            scanf("%s",ord);
            if(ord[0]=='Q'){
                scanf("%d",&u);
                printf("%d\n",query(1,1,n,vis[u]));
            }else{
                scanf("%d%d%d",&u,&v,&ww);
                if(ord[0]=='D')ww=-ww;
                update(u,v,ww);
            }
        }
    }
    return 0;
}
/*
3 2 5
1 2 3
2 1
2 3
I 1 3 5
Q 2
D 1 2 2
Q 1 
Q 3
--------
7
4
8
*/

---

因