hdu 3966 Aragorn's Story

树链剖分模板题。更新一条链的值以及求一个点的值。

 

可以用线段树或树状数组来区间更新加单点查询。

 

#include <bits/stdc++.h>
#define Lson l,mid,rt<<1
#define Rson mid+1,r,rt<<1|1
using namespace std;
const int M = 5e4+7;
int head[M],sz[M],son[M],f[M],id[M],rnk[M],dep[M],top[M];
int tot,cnt,n,m,q,a[M],pos,ans;
struct edge{
    int v,next;
}e[M<<1];
struct tree
{
    int sum,lazy,l,r;
}tree[M<<2];
void init(){
    tot=cnt=0;memset(head,-1,sizeof(head));
}
void add(int u,int v){
    e[++cnt].v=v;e[cnt].next=head[u];
    head[u]=cnt;
}
void dfs(int u,int fa,int d){
    sz[u]=1,son[u]=-1,f[u]=fa,dep[u]=d;
    for(int i=head[u];~i;i=e[i].next){
        int v=e[i].v;
        if(v==fa) continue;
        dfs(v,u,d+1);
        sz[u]+=sz[v];
        if(son[u]==-1||sz[v]>sz[son[u]]) son[u]=v;
    }
    return ;
}
void dfs1(int u,int t){
    id[u]=++tot;
    rnk[tot]=u;
    top[u]=t;
    if(son[u]==-1) return ;
    dfs1(son[u],t);
    for(int i=head[u];~i;i=e[i].next){
        int v=e[i].v;
        if(v==f[u]||v==son[u]) continue;
        dfs1(v,v);
    }
    return ;
}
void Pushup(int rt){
    tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void Pushdown(int rt){
    if(tree[rt].lazy){
        int v=tree[rt].lazy;
        tree[rt].lazy=0;
        tree[rt<<1].lazy+=v;
        tree[rt<<1|1].lazy+=v;
        tree[rt<<1].sum+=v*(tree[rt<<1].r-tree[rt<<1].l+1);
        tree[rt<<1|1].sum+=v*(tree[rt<<1|1].r-tree[rt<<1|1].l+1);
    }
}
void build(int l,int r,int rt){
    tree[rt].l=l;tree[rt].r=r;tree[rt].lazy=0;
    if(l==r){
        tree[rt].sum=a[rnk[l]];
        return ;
    }
    int mid=(l+r)>>1;
    build(Lson);
    build(Rson);
    Pushup(rt);
}
void update(int L,int R,int l,int r,int rt,int v){
    if(L<=l&&r<=R){
        tree[rt].lazy+=v;
        tree[rt].sum+=v*(r-l+1);
        return ;
    }
    Pushdown(rt);
    int mid=(l+r)>>1;
    if(L<=mid) update(L,R,Lson,v);
    if(R>mid) update(L,R,Rson,v);
    Pushup(rt);
}
void query(int l,int r,int rt){
    if(l==r){
        ans=tree[rt].sum;
        return ;
    }
    Pushdown(rt);
    int mid=(l+r)>>1;
    if(pos<=mid) query(Lson);
    if(pos>mid) query(Rson);
    return ;
}
void updates(int x,int y,int v){
    int fx=top[x],fy=top[y];
    while(fx!=fy){
        if(dep[fx]>dep[fy]){
            update(id[fx],id[x],1,n,1,v);
            x=f[fx],fx=top[x];
        }
        else{
            update(id[fy],id[y],1,n,1,v);
            y=f[fy],fy=top[y];
        }
    }
    if(dep[x]<dep[y])
        update(id[x],id[y],1,n,1,v);
    else
        update(id[y],id[x],1,n,1,v);
}
int main(){
    while(~scanf("%d%d%d",&n,&m,&q)){
        init();
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=1;i<=m;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);add(v,u);
        }
        dfs(1,-1,1);
        dfs1(1,1);
        build(1,n,1);
        while(q--){
            char c[2];
            int c1,c2,k;
            scanf("%s",c);
            if(c[0]=='I'||c[0]=='D'){
                scanf("%d%d%d",&c1,&c2,&k);
                if(c[0]=='D') k*=-1;
                updates(c1,c2,k);
            }
            else{
                scanf("%d",&c1);pos=id[c1];query(1,n,1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}