题目描述
有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个
操作,分为三种:
操作 1 :把某个节点 x 的点权增加 a 。
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 :询问某个节点 x 到根的路径中所有点的点权和。
解题思路
树链剖分的裸题。
更改一棵子树的值,就是在DFS序上更改一段区间。
询问时,每条重链其实也可以看成一段区间,构造DFS先遍历重儿子就可以了。
所以线段树维护就可以了。
#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=100005,maxm=200005;
int nxt[maxm],son[maxm],lnk[maxn],f[maxn],s[maxn],w[maxn],h[maxn],dep[maxn],id[maxn],who[maxn],top[maxn];
int n,Q,tot;
struct jz{
int L,R;
LL sum,tag;
}a[maxn*4];
inline int _read(){
int num=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
return num*f;
}
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void dfs1(int x){
s[x]=1;dep[x]=dep[f[x]]+1;
for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=f[x]){
f[son[j]]=x;dfs1(son[j]);
if (s[son[j]]>s[h[x]]) h[x]=son[j];
s[x]+=s[son[j]];
}
}
void dfs2(int x,int lst){
id[x]=++tot;who[tot]=x;top[x]=lst;
if (h[x]!=0) dfs2(h[x],lst);
for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=f[x]&&son[j]!=h[x]) dfs2(son[j],son[j]);
}
void Build(int x,int L,int R){
a[x].L=L;a[x].R=R;
if (L==R){a[x].sum=w[who[L]];return;}
int mid=L+(R-L>>1);
Build(x*2,L,mid);Build(x*2+1,mid+1,R);
a[x].sum=a[x*2].sum+a[x*2+1].sum;
}
void Pushdown(int x){
if (a[x].L==a[x].R) return;
LL tag=a[x].tag;a[x].tag=0;
a[x*2].tag+=tag;a[x*2+1].tag+=tag;
a[x*2].sum+=tag*(a[x*2].R-a[x*2].L+1);
a[x*2+1].sum+=tag*(a[x*2+1].R-a[x*2+1].L+1);
}
void change(int x,int L,int R,int y){
Pushdown(x);
if (L==a[x].L&&R==a[x].R){a[x].sum+=(LL)y*(R-L+1);a[x].tag=y;return;}
int mid=a[x].L+(a[x].R-a[x].L>>1);
if (R<=mid) change(x*2,L,R,y);else
if (L>mid) change(x*2+1,L,R,y);else
{change(x*2,L,mid,y);change(x*2+1,mid+1,R,y);}
a[x].sum=a[x*2].sum+a[x*2+1].sum;
}
LL query(int x,int L,int R){
Pushdown(x);
if (L==a[x].L&&R==a[x].R) return a[x].sum;
int mid=a[x].L+(a[x].R-a[x].L>>1);
if (R<=mid) return query(x*2,L,R);else
if (L>mid) return query(x*2+1,L,R);else
return query(x*2,L,mid)+query(x*2+1,mid+1,R);
a[x].sum=a[x*2].sum+a[x*2+1].sum;
}
void ask(int x,int y){
LL num=0;
while (top[x]!=top[y]){
if (dep[top[x]]<dep[top[y]]) swap(x,y);
num+=query(1,id[top[x]],id[x]);
x=f[top[x]];
}
if (dep[x]>dep[y]) swap(x,y);
num+=query(1,id[x],id[y]);
printf("%lld\n",num);
}
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
n=_read();Q=_read();
for (int i=1;i<=n;i++) w[i]=_read();
for (int i=1;i<n;i++){
int x=_read(),y=_read();
add(x,y);add(y,x);
}
dfs1(1);tot=0;dfs2(1,1);
Build(1,1,n);
while (Q--){
int x=_read(),y=_read(),z;
if (x==1) z=_read(),change(1,id[y],id[y],z);else
if (x==2) z=_read(),change(1,id[y],id[y]+s[y]-1,z);else
if (x==3) ask(1,y);
}
return 0;
}