[FROM WOJ]#2897 「HAOI2015」树上操作

#2897 「HAOI2015」树上操作

题面
有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：
1：把某个节点 x 的点权增加 a 。
2：把某个节点 x 为根的子树中所有点的点权都增加 a 。
3：询问某个节点 x 到根的路径中所有点的点权和。

输入
第一行包含两个整数 N,M。表示点数和操作数。
接下来一行 N 个整数，表示树中节点的初始权值。
接下来 N−1 行每行两个正整数 fr,to ， 表示该树中存在一条边 (fr,to)
再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操作的种类（1-3） ，之后接这个操作的参数（x 或者 x a） 。

输出
对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

样例输入
5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

样例输出
6
9
13

提示
对于 100%的数据， $N,M\le 100000$，且所有输入数据的绝对值都不会超过 $10^6$ 。

SOL
树链剖分的模板题。
按轻重链剖分之后，转成一般的区间问题，然后用线段树乱搞。
对于1、2两个操作，可以用一个change函数解决，都转化成区间修改的操作来处理。询问就按正常的方法写。

代码：

#include<bits/stdc++.h>
#define N 100005
#define int long long
using namespace std;
inline int rd(){
	int data=0,w=1;static char ch=0;
	while(!isdigit(ch)&&ch!='-')ch=getchar();
	if(ch=='-')w=-1,ch=getchar();
	while(isdigit(ch))data=(data<<1)+(data<<3)+ch-'0',ch=getchar();
	return data*w;
}
int n,m,ans,ins;
int cnt,first[N];
struct node{int v,nxt;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;}
int w[N],W[N],seg[N<<1],siz[N],son[N],f[N],dep[N],top[N];
void dfs1(int u,int fa,int d){
	siz[u]=1;f[u]=fa;dep[u]=d;
	for(int register i=first[u];i;i=e[i].nxt){
		int register v=e[i].v;
		if(v==fa)continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];if(siz[son[u]]<siz[v])son[u]=v;
	}
}
void dfs2(int u,int tp){
	seg[u]=++seg[0];
	w[seg[0]]=W[u];
	top[u]=tp;
	if(!son[u])return;	
	dfs2(son[u],tp);
	for(int register i=first[u];i;i=e[i].nxt){
		int register v=e[i].v;
		if(v==f[u]||v==son[u])continue;
		dfs2(v,v);
	}
}
#define lc (p<<1)
#define rc (p<<1|1)
struct segment_tree{int sum,add;}t[N<<2];
inline void pushup(int p){t[p].sum=t[lc].sum+t[rc].sum;}
inline void pushdown(int p,int l){
	if(!t[p].add)return;
	t[lc].add+=t[p].add;t[rc].add+=t[p].add;
	t[lc].sum+=t[p].add*(l-(l>>1));t[rc].sum+=t[p].add*(l>>1);
	t[p].add=0;
}
void build(int p,int l,int r){
	if(l==r){t[p].sum=w[l];return;}
	int register mid=l+r>>1;
	build(lc,l,mid);
	build(rc,mid+1,r);
	pushup(p);
}
void change(int p,int l,int r,int v,int ql,int qr){
	if(ql<=l&&qr>=r){t[p].sum+=v*(r-l+1);t[p].add+=v;return;}
	int register mid=l+r>>1;
	pushdown(p,r-l+1);
	if(ql<=mid)change(lc,l,mid,v,ql,qr);
	if(qr>mid)change(rc,mid+1,r,v,ql,qr);
	pushup(p);
}
void query(int p,int l,int r,int ql,int qr){
	if(ql<=l&&qr>=r){ans+=t[p].sum;return;}
	int mid=l+r>>1;
	pushdown(p,r-l+1);
	if(ql<=mid)query(lc,l,mid,ql,qr);
	if(qr>mid)query(rc,mid+1,r,ql,qr);
}
void ret(int x,int y){
	while(f[top[x]]!=f[top[y]]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		query(1,1,n,seg[top[x]],seg[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	query(1,1,n,seg[x],seg[y]);
}
signed main(){
	n=rd();m=rd();
	for(int register i=1;i<=n;i++)W[i]=rd();
	for(int register i=1;i<n;i++){int register fr=rd(),to=rd();add(fr,to);add(to,fr);}
	dfs1(1,0,1);dfs2(1,1);build(1,1,n);
	while(m--){ins=rd();
		if(ins==1){int register x=rd(),v=rd();change(1,1,n,v,seg[x],seg[x]);}
		if(ins==2){int register x=rd(),v=rd();change(1,1,n,v,seg[x],seg[x]+siz[x]-1);}
		if(ins==3){int register x=rd();ans=0;ret(1,x);printf("%lld\n",ans);}
	}
	return 0;
}