P3178 树链剖分 线段树 子树加

题目
输入样例#1：
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
输出样例#1：
6
9
13
说明
对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6

因为子树的dfs序连续，所以直接区间加pos[u]+siz[u]-1，便可实现子树加;

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+6;
int fa[N],son[N],dep[N],top[N],siz[N],pos[N];
ll fp[N];
struct Edge{
	int to,next;
	Edge(){}
	Edge(int to,int next):to(to),next(next){}
}E[N];
int head[N],tot;
inline void addedge(int u,int v){
	E[tot]=Edge(v,head[u]);
	head[u]=tot++;
}
void dfs(int x,int f,int d){
	siz[x]=1,fa[x]=f,dep[x]=d;
	for(int i=head[x];~i;i=E[i].next)if(E[i].to!=f){
		int v=E[i].to;
		dfs(v,x,d+1);
		siz[x]+=siz[v];
		if(son[v]==-1||siz[v]>siz[son[x]])son[x]=v;
	}
}
int cnt;
ll a[N];
void dfs2(int x,int nowtop){
	pos[x]=++cnt,top[x]=nowtop,fp[cnt]=a[x];
	if(son[x]==-1)return;
	dfs2(son[x],nowtop);
	for(int i=head[x];~i;i=E[i].next){
		int v=E[i].to;
		if(v!=fa[x]&&v!=son[x])dfs2(v,v);
	}
}
namespace segtree{
	#define ls x<<1
	#define rs x<<1|1
	struct node{
		int l,r,siz;
		ll sum,lazy;
		node(){}
		node(int l,int r):l(l),r(r){siz=r-l+1,sum=lazy=0;}
		void update(ll x){
			lazy+=x;
			sum+=1ll*siz*x;
		}
	}T[N<<2];
	void pushup(int x){
		T[x].sum=T[ls].sum+T[rs].sum;
	}
	void pushdown(int x){
		ll lazy=T[x].lazy;
		if(lazy){
			T[ls].update(lazy);
			T[rs].update(lazy);
			T[x].lazy=0;
		}
	}
	void build(int x,int l,int r){
		T[x]=node(l,r);
		if(l==r){
			T[x].sum=fp[l];
			return;
		}
		int mid=(l+r)>>1;
		build(ls,l,mid);
		build(rs,mid+1,r);
		pushup(x);
	}
	void update(int x,int L,int R,int k) {
		int l=T[x].l,r=T[x].r;
		if(L<=l&&r<=R){
			T[x].update(k);return;
		}
		pushdown(x);
		int mid=(l+r)>>1;
		if(L<=mid)update(ls,L,R,k);
		if(R> mid)update(rs,L,R,k);
		pushup(x);
	}
	ll query(int x,int L,int R){
		int l=T[x].l,r=T[x].r;
		if(L<=l&&r<=R)return T[x].sum;
		pushdown(x);
		int mid=(l+r)>>1;
		ll ans=0;
		if(L<=mid)ans+=query(ls,L,R);
		if(R> mid)ans+=query(rs,L,R);
		return ans;
	}
}using namespace segtree;
ll query(int u,int v){
	int f1=top[u],f2=top[v];
	ll ans=0;
	while(f1^f2){
		if(dep[f1]<dep[f2])swap(f1,f2),swap(u,v);
		ans+=query(1,pos[f1],pos[u]);
		u=fa[f1],f1=top[u];
	}
	if(dep[u]>dep[v])swap(u,v);
	return ans+query(1,pos[u],pos[v]);
}
void init(){
	tot=0,memset(head,-1,sizeof(head));
	cnt=0,memset(son,-1,sizeof(son));
}
int u,v,q,op,n;
int main(){
	init();
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;++i)scanf("%lld",&a[i]);
	for(int i=1;i<n;++i)scanf("%d%d",&u,&v),addedge(u,v),addedge(v,u);
	dfs(1,0,0);
	dfs2(1,1);
	build(1,1,cnt);
	while(q--){
		scanf("%d%d",&op,&u);
		if(op==1)scanf("%d",&v),update(1,pos[u],pos[u],v);
		if(op==2)scanf("%d",&v),update(1,pos[u],pos[u]+siz[u]-1,v);
		if(op==3)printf("%lld\n",query(1,u));
	}
	return 0;
}