#include <bits/stdc++.h>
using namespace std;
#define ls l,mid,rt*2
#define rs mid+1,r,rt*2+1
#define mi (l+r)/2
const int MAXN=5e4+7;
const int INF=0x3f3f3f3f;
vector <int> edge[MAXN];
vector <int> linkk[MAXN];
int n,m,q;
int a[MAXN];
int tsize[MAXN],dep[MAXN],hson[MAXN],father[MAXN];//dfs1
int order[MAXN],transpos[MAXN];
int belong[MAXN],top[MAXN],cnt;//dfs2
int tree[MAXN*4];
long long lazy[MAXN*4];
int st,en;
void ini(){
for(int i=1;i<=n;i++){
edge[i].clear();
belong[i]=0;
}
linkk[1].clear();
dep[0]=-1;
tsize[0]=0;cnt=0;top[1]=1;
}
void dfs1(int now,int fa){
father[now]=fa;
int len=edge[now].size();
dep[now]=dep[fa]+1;
tsize[now]=1;
hson[now]=0;
for(int i=0;i<len;i++){
int pos=edge[now][i];
if(pos!=fa){
dfs1(pos,now);
tsize[now]+=tsize[pos];
if(tsize[pos]>tsize[hson[now]]) hson[now]=pos;
}
}
return ;
}
void dfs2(int now,int bel,int fa){
belong[now]=bel;
linkk[bel].push_back(now);
int len=edge[now].size();
if(hson[now]!=0)
dfs2(hson[now],bel,now);
for(int i=0;i<len;i++){
int pos=edge[now][i];
if(pos!=now&&pos!=fa&&pos!=hson[now]){
cnt++;
linkk[cnt].clear();
top[cnt]=pos;
dfs2(pos,cnt,now);
}
}
}
void push_down(int rt){
if(lazy[rt]!=0){
lazy[rt*2]+=lazy[rt];
lazy[rt*2+1]+=lazy[rt];
lazy[rt]=0;
}
}
void build(int l,int r,int rt){
lazy[rt]=0;
if(l==r){
tree[rt]=a[order[l]];
return ;
}
int mid=mi;
build(ls);build(rs);
}
void update(int l,int r,int rt,int v){
if(l>en||r<st) return ;
if(st<=l&&r<=en){
lazy[rt]+=v;
return ;
}
push_down(rt);
int mid=mi;
update(ls,v);update(rs,v);
}
long long query(int l,int r,int rt){
if(l==r){
return lazy[rt]+tree[rt];
}
push_down(rt);
int mid=mi;
return st<=mid?query(ls):query(rs);
}
void solveU(int x,int y,int v){
while(1){
if(top[belong[x]]==top[belong[y]]){
if(dep[x]>dep[y]) swap(x,y);
st=transpos[x];en=transpos[y];
update(1,n,1,v);
break;
}else{
if(dep[top[belong[x]]]>dep[top[belong[y]]]) swap(x,y);
st=transpos[top[belong[y]]];en=transpos[y];
update(1,n,1,v);
y=father[top[belong[y]]];
}
}
}
int main()
{
while(scanf("%d%d%d",&n,&m,&q)!=-1){int x,y;
ini();
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<n;i++){
scanf("%d%d",&x,&y);
edge[x].push_back(y);
edge[y].push_back(x);
}
dfs1(1,0);
dfs2(1,++cnt,0);
int ppp=0;
for(int i=1;i<=cnt;i++){
int len=linkk[i].size();
for(int j=0;j<len;j++){
order[++ppp]=linkk[i][j];
transpos[linkk[i][j]]=ppp;
}
}
build(1,n,1);
while(q--){char op[5];int v;
scanf("%s",op);
if(op[0]=='I'){
scanf("%d%d%d",&x,&y,&v);
solveU(x,y,v);
}else if(op[0]=='D'){
scanf("%d%d%d",&x,&y,&v);
solveU(x,y,-v);
}else{
scanf("%d",&x);
st=en=transpos[x];
printf("%lld\n",query(1,n,1));
}
}
}
}
树剖自用模板
最新推荐文章于 2020-12-07 20:11:07 发布