【bzoj 4034】 树上操作 【HAOI2015】

Description

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个

操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

Input

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 

行每行三个正整数 fr, to ， 表示该树中存在一条边 (fr, to) 。再接下来 M 行，每行分别表示一次操作。其中

第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

Output

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

Sample Input

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

Sample Output

6
9
13

HINT

 

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

这道题是一个树链剖分，对于2操作，只要记录以当前节点为根的子树中最后被访问的点即可，下面是程序：

#include<stdio.h>
#include<iostream>
#define ll long long
#define ci const int
using namespace std;
const int N=100005;
struct treenode{
	int lc,rc;
	ll w,c;
};
struct Segment_tree{
	treenode t[N<<2];
	int k;
	Segment_tree(){
		k=1;
	}
	void build(ci rt,ci l,ci r){
		if(l==r){
			t[rt].lc=t[rt].rc=0;
			t[rt].w=t[rt].c=0;
			return;
		}
		ci m=l+r>>1;
		build(t[rt].lc=++k,l,m);
		build(t[rt].rc=++k,m+1,r);
	}
	void add(ci rt,ci l,ci r,ci a,ci b,ll w){
		t[rt].w+=(b-a+1)*w;
		if(l==a&&r==b){
			t[rt].c+=w;
			return;
		}
		ci m=l+r>>1;
		if(b<=m){
			add(t[rt].lc,l,m,a,b,w);
		}
		else{
			if(a>m){
				add(t[rt].rc,m+1,r,a,b,w);
			}
			else{
				add(t[rt].lc,l,m,a,m,w);
				add(t[rt].rc,m+1,r,m+1,b,w);
			}
		}
	}
	ll ask(ci rt,ci l,ci r,ci a,ci b,const ll c){
		if(l==a&&r==b){
			return t[rt].w+c*(r-l+1);
		}
		ci m=l+r>>1;
		if(b<=m){
			return ask(t[rt].lc,l,m,a,b,c+t[rt].c);
		}
		if(a>m){
			return ask(t[rt].rc,m+1,r,a,b,c+t[rt].c);
		}
		return ask(t[rt].lc,l,m,a,m,c+t[rt].c)+ask(t[rt].rc,m+1,r,m+1,b,c+t[rt].c);
	}
}t;
struct edge{
	int v,next;
}e[N<<1];
int a[N],head[N],n,k,sum[N],dfn[N],son[N],dep[N],top[N],fa[N],last[N],dfs_clock;
void add(int u,int v){
	e[++k]=(edge){v,head[u]};
	head[u]=k;
}
void dfs1(int u,int f,int d){
	fa[u]=f;
	sum[u]=1;
	dep[u]=d;
	son[u]=0;
	int i;
	for(i=head[u];i;i=e[i].next){
		if(e[i].v!=f){
			dfs1(e[i].v,u,d+1);
			sum[u]+=sum[e[i].v];
			if(sum[e[i].v]>sum[son[u]]){
				son[u]=e[i].v;
			}
		}
	}
}
void dfs2(int u,int t){
	top[u]=t;
	last[u]=dfn[u]=++dfs_clock;
	if(!son[u]){
		return;
	}
	dfs2(son[u],t);
	last[u]=max(last[u],last[son[u]]);
	int i;
	for(i=head[u];i;i=e[i].next){
		if(e[i].v!=fa[u]&&e[i].v!=son[u]){
			dfs2(e[i].v,e[i].v);
			last[u]=max(last[u],last[e[i].v]);
		}
	}
}
ll ask(int x){
	ll ans=0;
	while(top[x]!=1){
		ans+=t.ask(1,1,n,dfn[top[x]],dfn[x],0);
		x=fa[top[x]];
	}
	return ans+t.ask(1,1,n,1,dfn[x],0);
}
int main(){
	int m,i,x,w;
	int cnt=0;
	scanf("%d%d",&n,&m);
	t.build(1,1,n);
	for(i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(i=1;i<n;i++){
		scanf("%d%d",&x,&w);
		add(x,w);
		add(w,x);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	for(i=1;i<=n;i++){
		t.add(1,1,n,dfn[i],dfn[i],a[i]);
	}
	while(m--){
		scanf("%d%d",&i,&x);
		if(m==605){
			cnt=1;
		}
		switch(i){
			case 1:
				scanf("%d",&w);
				t.add(1,1,n,dfn[x],dfn[x],w);
				break;
			case 2:
				scanf("%d",&w);
				t.add(1,1,n,dfn[x],last[x],w);
				break;
			case 3:
				printf("%lld\n",ask(x));
				break;
		}
	}
	return 0;
}