Luogu P3178 [HAOI2015]树上操作

https://www.luogu.org/problemnew/show/P3178

题目描述

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

输入输出格式

输入格式：

 

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

 

输出格式：

 

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

 

输入输出样例

输入样例#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 。

 

这是一道树剖模板题

#include<cstdio>
#include<iostream>
#define ll long long
using namespace std;

const int N=1e6+5;
int n,m,a[N],b[N];
int cnt,to[N],nxt[N],he[N];
int dad[N],siz[N],son[N],dep[N],id[N],top[N],tot;

inline void add(int u,int v)
{
	to[++cnt]=v,nxt[cnt]=he[u],he[u]=cnt;
}

void dfs1(int fa,int u)
{
	dad[u]=fa,dep[u]=dep[fa]+1,siz[u]=1;
	int mx=0;
	for(int e=he[u];e;e=nxt[e])
	{
		int v=to[e];
		if(v!=fa) 
		{
			dfs1(u,v); siz[u]+=siz[v];
			if(siz[v]>mx) son[u]=v,mx=siz[v];
		}
	}
}

void dfs2(int fa,int u,int t)
{
	top[u]=t,id[u]=++tot;
	if(!son[u]) return;
	dfs2(u,son[u],t);
	for(int e=he[u];e;e=nxt[e])
	{
		int v=to[e];
		if(v!=fa&&v!=son[u]) dfs2(u,v,v);
	}
}

struct A
{
	ll c[N],t[N];
	inline void up(int p)
	{
		c[p]=c[p<<1]+c[p<<1|1];
	}
	
	inline void down(int p,int l,int r,int mid)
	{
		if(l<r)
		{
			c[p<<1]+=t[p]*(mid-l+1),t[p<<1]+=t[p];
			c[p<<1|1]+=t[p]*(r-mid),t[p<<1|1]+=t[p];	
		} 
		t[p]=0;
	}
	
	void build(int p,int l,int r)
	{
		if(l==r) 
		{
			c[p]=b[l]; return;
		}
		int mid=l+r>>1;
		build(p<<1,l,mid),build(p<<1|1,mid+1,r);
		up(p);
	}
	
	void add(int p,int l,int r,int x,int y,int k)
	{
		if(l==x&&r==y)
		{
			t[p]+=k,c[p]+=(ll)(r-l+1)*k;
			return;
		}
		int mid=l+r>>1;
		down(p,l,r,mid);
		if(mid>=y) add(p<<1,l,mid,x,y,k);
			else if(mid<x) add(p<<1|1,mid+1,r,x,y,k);
				else add(p<<1,l,mid,x,mid,k),add(p<<1|1,mid+1,r,mid+1,y,k);
		up(p);
	}
	
	ll sum(int p,int l,int r,int x,int y)
	{
		if(l==x&&r==y) return c[p];
		int mid=l+r>>1;
		down(p,l,r,mid);
		if(mid>=y) return sum(p<<1,l,mid,x,y);
			else if(mid<x) return sum(p<<1|1,mid+1,r,x,y);
				else return sum(p<<1,l,mid,x,mid)+sum(p<<1|1,mid+1,r,mid+1,y);
	}
}tree;

ll sum(int u,int v)
{
	ll ret=0;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]>dep[top[v]]) swap(u,v);
		ret+=tree.sum(1,1,n,id[top[v]],id[v]);
		v=dad[top[v]];
	}
	if(id[u]>id[v]) swap(u,v);
	ret+=tree.sum(1,1,n,id[u],id[v]);
	return ret;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<n;i++) 
	{
		int u,v; scanf("%d%d",&u,&v);
		add(u,v),add(v,u);
	}
	dfs1(0,1); dfs2(0,1,1);
	for(int i=1;i<=n;i++) b[id[i]]=a[i];
	tree.build(1,1,n);
	while(m--)
	{
		int t; scanf("%d",&t);
		if(t==1)
		{
			int x,y; scanf("%d%d",&x,&y);
			tree.add(1,1,n,id[x],id[x],y);
		}else
		if(t==2)
		{
			int x,y; scanf("%d%d",&x,&y);
			tree.add(1,1,n,id[x],id[x]+siz[x]-1,y);
		}else
		{
			int x; scanf("%d",&x);
			printf("%lld\n",sum(x,1));
		}
	}
	return 0;
}