NEFU 1160 线段树

#pragma comment(linker, "/STACK:102400000, 102400000")
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=100010;
vector<int>G[maxn];
ll num[maxn*4],lazy[maxn*4],min1[maxn*4];
int L[maxn],R[maxn];
int idx;
void dfs(int u,int f){
L[u]=++idx;
for(unsigned int i=0;i<G[u].size();i++){
int v=G[u][i];
if(v==f) continue;
dfs(v,u);
}
R[u]=idx;
}
void pushup(int node){
num[node]=num[node<<1]+num[node<<1|1];
min1[node]=min(min1[node<<1],min1[node<<1|1]);
}
void pushdown(int node,int m){
if(lazy[node]){
lazy[node<<1]+=lazy[node];
lazy[node<<1|1]+=lazy[node];
num[node<<1]+=lazy[node]*(ll)(m-(m>>1));
num[node<<1|1]+=lazy[node]*(ll)(m>>1);
min1[node<<1]+=lazy[node];
min1[node<<1|1]+=lazy[node];
lazy[node]=0;
}
}
void buildtree(int le,int ri,int node){
if(le==ri){
min1[node]=num[node]=1;
return ;
}
int t=(le+ri)>>1;
buildtree(le,t,node<<1);
buildtree(t+1,ri,node<<1|1);
pushup(node);
}
void update(int l,int r,ll x,int le,int ri,int node){
if(l<=le&&ri<=r){
lazy[node]+=x;
num[node]+=x*(ll)(ri-le+1);
min1[node]+=x;
return ;
}
pushdown(node,ri-le+1);
int t=(le+ri)>>1;
if(l<=t) update(l,r,x,le,t,node<<1);
if(r>t) update(l,r,x,t+1,ri,node<<1|1);
pushup(node);
}
ll query_sum(int l,int r,int le,int ri,int node){
if(l<=le&&ri<=r) return num[node];
pushdown(node,ri-le+1);
ll ans=0;
int t=(le+ri)>>1;
if(l<=t) ans+=query_sum(l,r,le,t,node<<1);
if(r>t) ans+=query_sum(l,r,t+1,ri,node<<1|1);
return ans;
}
ll query_min(int l,int r,int le,int ri,int node){
if(l<=le&&ri<=r) return min1[node];
pushdown(node,ri-le+1);
ll ans=999999999999ll;
int t=(le+ri)>>1;
if(l<=t) ans=min(ans,query_min(l,r,le,t,node<<1));
if(r>t) ans=min(ans,query_min(l,r,t+1,ri,node<<1|1));
return ans;
}
int main(){
int T,cas=1,n,m,a,b,c;
char ch[10];
scanf("%d",&T);
while(T--){
scanf("%d",&n);
for(int i=0;i<maxn;i++) G[i].clear();
for(int i=0;i<n-1;i++){
scanf("%d%d",&a,&b);
G[a].push_back(b);
}
idx=0;dfs(1,0);
buildtree(1,n,1);
memset(lazy,0,sizeof(lazy));
scanf("%d",&m);
printf("Case #%d:\n",cas++);
while(m--){
scanf("%s",&ch);
if(ch[0]=='C'){
scanf("%d%d",&b,&c);
ll kk=(ll)c;
ll minmin=query_min(L[b],R[b],1,n,1);
if(c+minmin<0) kk=-minmin;
update(L[b],R[b],kk,1,n,1);
}else if(ch[0]=='Q'){
scanf("%d",&b);
ll answe=query_sum(L[b],R[b],1,n,1);
printf("%lld\n",answe);
}
}
}
return 0;
}