大概是一道裸题。注意处理给所有子树add时可以采用给入时间和出时间的区域线段树统一加add的方法,因为子树包含且仅包含在这一段区域内。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int maxn=100005;
struct edge
{
int to,next;
}e[maxn<<1];
struct tree
{
long long val,add;
}t[maxn<<2];
int n,q,dfs_clock,cnt;
int s[maxn],f[maxn],id[maxn],out_time[maxn],head[maxn];
int size[maxn],son[maxn],per[maxn],top[maxn];
inline void insert(int a,int b)
{
e[++cnt].to=b;e[cnt].next=head[a];head[a]=cnt;
}
inline void maintain(int ro)
{
t[ro].val=t[ro<<1].val+t[ro<<1|1].val;
}
void build(int ro,int l,int r)
{
if(l==r)
{
t[ro].val=s[f[l]];
return;
}
int mid=l+r>>1;
build(ro<<1,l,mid);
build(ro<<1|1,mid+1,r);
maintain(ro);
}
inline void pushdown(int ro,int l,int r)
{
int mid=l+r>>1;
if(t[ro].add)
{
t[ro<<1].add+=t[ro].add;
t[ro<<1|1].add+=t[ro].add;
t[ro<<1].val+=1LL*(mid-l+1)*t[ro].add;
t[ro<<1|1].val+=1LL*(r-mid)*t[ro].add;
t[ro].add=0;
}
}
void addition(int ro,int L,int R,int l,int r,int add)
{
if(L==l&&R==r)
{
t[ro].val+=1LL*(r-l+1)*add;
t[ro].add+=add;
return;
}
pushdown(ro,l,r);
int mid=l+r>>1;
if(R<=mid)addition(ro<<1,L,R,l,mid,add);
else if(L>=mid+1)addition(ro<<1|1,L,R,mid+1,r,add);
else addition(ro<<1,L,mid,l,mid,add),addition(ro<<1|1,mid+1,R,mid+1,r,add);
maintain(ro);
}
long long query(int ro,int L,int R,int l,int r)
{
if(L==l&&R==r)
return t[ro].val;
pushdown(ro,l,r);
int mid=l+r>>1;
if(R<=mid)return query(ro<<1,L,R,l,mid);
else if(L>=mid+1)return query(ro<<1|1,L,R,mid+1,r);
else return query(ro<<1,L,mid,l,mid)+query(ro<<1|1,mid+1,R,mid+1,r);
}
void addition(int x,int add)
{
addition(1,id[x],out_time[x],1,n,add);
}
long long query(int x)
{
long long res=0;
int tp=top[x];
while(x)
{
res+=query(1,id[tp],id[x],1,n);
x=per[tp];
tp=top[x];
}
return res;
}
void dfs(int x,int fa)
{
size[x]=1;
per[x]=fa;
for(int i=head[x];i;i=e[i].next)
if(e[i].to!=fa)
{
dfs(e[i].to,x);
size[x]+=size[e[i].to];
if(size[e[i].to]>size[son[x]])
son[x]=e[i].to;
}
}
void dfs_chain(int x,int tp)
{
id[x]=++dfs_clock;//节点映射编号
f[dfs_clock]=x;//编号映射节点
top[x]=tp;
if(son[x])
dfs_chain(son[x],tp);
for(int i=head[x];i;i=e[i].next)
if(e[i].to!=per[x]&&e[i].to!=son[x])
dfs_chain(e[i].to,e[i].to);
out_time[x]=dfs_clock;
}
int main()
{
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%d",s+i);
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
insert(a,b);
insert(b,a);
}
dfs(1,0);
dfs_chain(1,1);
build(1,1,n);
while(q--)
{
int d,d1,d2;
scanf("%d",&d);
if(d==1)
{
scanf("%d%d",&d1,&d2);
addition(1,id[d1],id[d1],1,n,d2);
}
else if(d==2)
{
scanf("%d%d",&d1,&d2);
addition(d1,d2);
}
else if(d==3)
{
scanf("%d",&d1);
printf("%lld\n",query(d1));
}
}
return 0;
}