Codeforces 383C Propagating tree, 线段树, 黑白染色思想

按深度染色，奇深度的点存反权值。

#include <bits/stdc++.h>
using namespace std;

vector <int> g[500005];
int t1,t2,t3,seq[500005],a[1000005],s[500005],vis[500005],ind,n,m,src[500005],frm[500005],dep[500005];

void dfs(int p) {
    vis[p]=1;
    frm[p]=ind+1;
    for(int i=0;i<g[p].size();i++)
        if(vis[g[p][i]]==0) 
            dep[g[p][i]]=dep[p]+1,
            dfs(g[p][i]);
    seq[p]=++ind;
}

void build(int p,int l,int r){
    if(l==r) a[p]=s[l];
    else build(p*2,l,(l+r)/2),build(p*2+1,(l+r)/2+1,r);
}

void modify(int p,int l,int r,int ql,int qr,int k){
    if(l>qr||r<ql) return;
    if(l>=ql&&r<=qr) {a[p]+=k; return;}
    a[p*2]+=a[p], a[p*2+1]+=a[p], a[p]=0;
    modify(p*2,l,(l+r)/2,ql,qr,k);
    modify(p*2+1,(l+r)/2+1,r,ql,qr,k);
}

int query(int p,int l,int r,int pos){
    if(l==r) return a[p];
    a[p*2]+=a[p], a[p*2+1]+=a[p], a[p]=0;
    if(pos<=(l+r)/2) return query(p*2,l,(l+r)/2,pos);
    else return query(p*2+1,(l+r)/2+1,r,pos);
}

int main(){
    ios::sync_with_stdio(false);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&src[i]);
    for(int i=1;i<n;i++) scanf("%d%d",&t1,&t2), g[t1].push_back(t2), g[t2].push_back(t1);
    dep[1]=1; dfs(1);
    for(int i=1;i<=n;i++) dep[i]=-1+2*(dep[i]%2);
    for(int i=1;i<=n;i++) s[seq[i]]=dep[i]*src[i];
    build(1,1,n);
    for(int i=1;i<=m;i++) {
        scanf("%d%d",&t1,&t2);
        if(t1==1){
            scanf("%d",&t3);
            modify(1,1,n,frm[t2],seq[t2],dep[t2]*t3);
        }
        else{
            printf("%d
",dep[t2]*query(1,1,n,seq[t2]));
        }
    }
    return 0;
}