CodeForces - 383C Propagating tree （DFS序+线段树）

#include <bits/stdc++.h>
using namespace std;
#define ls l,mid,rt*2,k
#define rs mid+1,r,rt*2+1,k
#define sf rt,k
#define mi (l+r)/2
const int MAXN=2e5+7;
vector <int> edge[MAXN];
int n,m,a[MAXN];
int order[2][MAXN],depth[MAXN],in[2][MAXN],out[2][MAXN],cnt[2];
int ppp[2],tree[2][MAXN*4],lazy[2][MAXN*4];
int st,en,v;
void dfs(int pos,int dep){
depth[pos]=dep;
int k=dep%2;
cnt[k]++;
order[k][cnt[k]]=pos;
in[k][pos]=cnt[k];
in[!k][pos]=cnt[!k]+1;
int len=edge[pos].size();
for(int i=0;i<len;i++) dfs(edge[pos][i],dep+1);
out[k][pos]=cnt[k];
out[!k][pos]=cnt[!k];
}
void push_down(int rt,int k){
if(lazy[k][rt]!=0){
lazy[k][rt*2]+=lazy[k][rt];
lazy[k][rt*2+1]+=lazy[k][rt];
}
lazy[k][rt]=0;
}
void build(int l,int r,int rt,int k){
lazy[k][rt]=0;
if(l==r){
tree[k][rt]=a[order[k][++ppp[k]]];
return ;
}
int mid=mi;
build(ls);
build(rs);
}
void update(int l,int r,int rt,int k){
if(l>en||r<st) return ;
if(st<=l&&r<=en){
lazy[k][rt]+=v;
return ;
}
int mid=mi;
push_down(sf);
update(ls);update(rs);
}
void query(int l,int r,int rt,int k){
if(l>en||r<st) return ;
if(l==r){
v=tree[k][rt]+lazy[k][rt];
return ;
}
int mid=mi;
push_down(sf);
query(ls);query(rs);
}
int main(){
int x,y,op;
while(scanf("%d%d",&n,&m)!=-1){
for(int i=1;i<=n;i++) scanf("%d",&a[i]),edge[i].clear();
for(int i=1;i<n;i++){
scanf("%d%d",&x,&y);
edge[x].push_back(y);
}
if(n==1){
while(m--){
scanf("%d",&op);
if(op==1){
scanf("%d%d",&x,&y);
a[x]+=y;
}else{
scanf("%d",&x);
printf("%d\n",a[x]);
}
}
}else{
cnt[0]=cnt[1]=0;
dfs(1,1);
ppp[0]=ppp[1]=0;
build(1,cnt[0],1,0);
build(1,cnt[1],1,1);
while(m--){
scanf("%d",&op);
if(op==1){
scanf("%d%d",&x,&y);
int k=depth[x]%2;
st=in[k][x],en=out[k][x],v=y;
update(1,cnt[k],1,k);
st=in[!k][x],en=out[!k][x],v=-y;
if(st<=en) update(1,cnt[!k],1,!k);
}else{
scanf("%d",&x);
int k=depth[x]%2;
st=en=in[k][x];
query(1,cnt[k],1,k);
printf("%d\n",v);
}
}
}
}
}