P3178 [HAOI2015]树上操作-树链剖分

有一棵点数为 N 的树，以点 1 为根，且树点有权。然后有 M 个操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

1. 这个题比较简单，操作3的根是固定的
2. 容易写错的地方，一个是树上的点x，要在线段树里一定是dfn[x]

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<ctime>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=100100;

inline int read(){
	char ch=' ';int f=1;int x=0;
	while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*f;
}

int n,m;
ll z[N];
struct node
{
	int v,nxt;
}edge[N<<1];
int head[N],cnt;

void add(int u,int v)
{
	cnt++;
	edge[cnt].v=v;
	edge[cnt].nxt=head[u];
	head[u]=cnt;	
}
int dep[N];
int siz[N];
int faz[N];
int son[N];
int top[N];
int dfn[N];
int tid[N];
int cc=0;

void dfs1(int u,int fa)
{
	faz[u]=fa;siz[u]=1;
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		if(v==fa) continue;
		
		dep[v]=dep[u]+1;
		dfs1(v,u);
		
		siz[u]+=siz[v];
		if(!son[u]||siz[v]>siz[son[u]]) son[u]=v;
	}
}
void dfs2(int u,int t)
{
	top[u]=t;
	cc++;
	dfn[u]=cc;
	tid[cc]=u;
	
	if(!son[u]) return ;
	dfs2(son[u],t);
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		if(v!=son[u]&&v!=faz[u])
		{
			dfs2(v,v);
		}
	}
}

ll s[N<<2];
ll t[N<<2];
void update(int rt)
{
	s[rt]=s[rt<<1]+s[rt<<1|1];
}
void build(int rt,int l,int r)
{
	if(l==r)
	{
		s[rt]=z[tid[l]];
		t[rt]=0;
		return ;
	}
	int mid=(l+r) >> 1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	update(rt);
}
void push_down(int rt,int l,int r)
{
		ll k=t[rt];
		int mid=(l+r)>>1;
		t[rt]=0;
		t[rt<<1]+=k;
		t[rt<<1|1]+=k;
		s[rt<<1]+=(mid-l+1)*k;
		s[rt<<1|1]+=(r-mid)*k;
}
void modify(int rt,int l,int r,int x,int y,ll k)
{
	if(l>=x&&r<=y)
	{
		s[rt]+=(r-l+1)*k;
		t[rt]+=k;
		return ;
	}
	push_down(rt,l,r);
	int mid=(l+r)>> 1;
	if(x<=mid)
	{
		modify(rt<<1,l,mid,x,y,k);
	}
	if(y>mid)
	{
		modify(rt<<1|1,mid+1,r,x,y,k);
	}
	update(rt);//
}
ll query(int rt,int l,int r,int x,int y)
{
	if(l>=x&&r<=y)
	{
		return s[rt];
	}
	push_down(rt,l,r);
	int mid=(l+r)>>1 ;
	ll ans=0;
	if(x<=mid)
	{
		ans+=query(rt<<1,l,mid,x,y);
	}
	if(y>mid)
	{
		ans+=query(rt<<1|1,mid+1,r,x,y);
	}
	return ans;
}
ll query_path(int x)
{
	ll ans=0;
	int fx=top[x];	
	while(fx!=1)
	{
		ans+=query(1,1,n,dfn[fx],dfn[x]);
		x=faz[fx];
		fx=top[x];
	}
	ans+=query(1,1,n,1,dfn[x]);
	return ans;
}
int main()
{
	n=read();m=read();
	for(int i=1;i<=n;i++)
	z[i]=read();
	
	int u,v;
	for(int i=1;i<=n-1;i++)
	{
		u=read();v=read();
		add(u,v);add(v,u);
	}
	dep[1]=1;
	dfs1(1,0);
	dfs2(1,1);
	
	build(1,1,n);
	
	ll op,x,a;
	for(int i=1;i<=m;i++)
	{
		op=read();
		if(op==1)
		{
			x=read();a=read();
			modify(1,1,n,dfn[x],dfn[x],a);
		}
		else if(op==2)
		{
			x=read();a=read();
			modify(1,1,n,dfn[x],dfn[x]+siz[x]-1,a);
		}
		else
		{
			x=read();
			printf("%lld\n",query_path(x));
		}
	}
	return 0;
}