「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≤100000，且所有输入数据的绝对值都不会超过 10^6 。
网上资源自己的理解 ：

#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){//初始化线段树w[1]到w[n]的值，建立top数组 
	seg[u]=++seg[0];//seg[0]保存seg数组长度（下标从1开始），该数组可由树上节点找到其在线段树上的对应位置 
	w[seg[0]]=W[u];//w数组保存线段树对应1到n的值
	top[u]=tp;//继承父亲的top值 
	if(!son[u])return;//u为叶子节点	
	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);//轻儿子的top值即为儿子本身 
	}
}
#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]]){//个人觉得还是改为top[x]!=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;
}

自己的尝试及错误分析：

#include<bits/stdc++.h>
#define mid (T[p].l+T[p].r>>1)
#define lc (p<<1)
#define rc (p<<1|1) 
#define int long long
using namespace std;
const int N=100005;
int read(){
	int s=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-f;
		ch=getchar();
	}
	while(isdigit(ch)){
		s=(s<<3)+(s<<1)+ch-'0';
		ch=getchar();
	}
	return s*f;
}
struct fjy{//线段树的相关信息记录 
	int l,r,lz,sum;
}T[N<<2];
struct lcz{//邻接链表 
	int nxt,v;
}z[N<<1];
int first[N<<1],cnt,w[N];
void add(int u,int v){
	z[++cnt].nxt=first[u];z[cnt].v=v;first[u]=cnt;
}
void pushup(int p){//更新T[p]的相关值 
	T[p].sum=T[lc].sum+T[rc].sum; return;
}
void pushnow(int p,int v){//修改区间T[p]的相关值（增加v） 
	T[p].sum+=v*(T[p].r-T[p].l+1);
	T[p].lz+=v;
	return;
}
void pushdown(int p){//下传T[p]的懒惰标记 
	if(T[p].lz==0) return;
	pushnow(lc,T[p].lz);
	pushnow(rc,T[p].lz);
	T[p].lz=0;
	return;
}
void build(int p,int l,int r){
	T[p].l=l,T[p].r=r;//printf("<%d>",T[1].l);
	if(l==r){
		T[p].sum=w[l];
		return;
	}
	build(lc,l,mid);build(rc,mid+1,r);
	pushup(p);
}
void update(int p,int ql,int qr,int v){
	if(T[p].l>qr||T[p].r<ql) return;
	if(T[p].l>=ql&&T[p].r<=qr) { pushnow(p,v);return; }
	pushdown(p);
	if(ql<=mid) update(lc,ql,qr,v);
	if(qr>mid) update(rc,ql,qr,v);
	pushup(p);
}
int query(int p,int ql,int qr){
	if(T[p].l>qr||T[p].r<ql) return 0;
	if(T[p].l>=ql&&T[p].r<=qr) { return T[p].sum; }
	pushdown(p);
	int ans=0;
	if(ql<=mid) ans+=query(lc,ql,qr);
	if(qr>mid) ans+=query(rc,ql,qr);//printf("ANSQ=%d",ans);
	return ans;
}
int dep[N],siz[N],son[N],f[N];
void dfs1(int u,int fa,int k){//第一次深搜，得到重儿子，子树大小 
	siz[u]=1;f[u]=fa;dep[u]=k;
	for(int i=first[u];i;i=z[i].nxt){
		if(z[i].v==fa) continue;
		dfs1(z[i].v,u,k+1);
		siz[u]+=siz[z[i].v];
		if(siz[son[u]]<siz[z[i].v]) son[u]=z[i].v;
	}
}
int seg[N],W[N],top[N];
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 i=first[u];i;i=z[i].nxt){
		if(z[i].v==f[u]||z[i].v==son[u]) continue;
		dfs2(z[i].v,z[i].v);
	}
}
void swap(int &x,int &y){
	int c=x;
	x=y;y=c;return;
}
int ask(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);//注意：不是dep[x]而是dep[top[x]]，且是小于号而不是大于符号，要选择深度大的节点向上跳 
		ans+=query(1,seg[top[x]],seg[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans+=query(1,seg[x],seg[y]);
	return ans;
}
/*int ask(int x,int y){
	int ans=0;
	while(f[top[x]]!=f[top[y]]){	
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		ans+=query(1,seg[top[x]],seg[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	ans+=query(1,seg[x],seg[y]);return ans;
}*/
int n,m,ui,vi,judge,xi,ai,ans;
signed main(){
	n=read();m=read();
	for(int i=1;i<=n;i++) W[i]=read();
	for(int i=1;i<n;i++){
		ui=read(),vi=read();
		add(ui,vi),add(vi,ui);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n);
	//printf("<>%d<>",T[4].sum);
	for(int i=1;i<=m;i++){
		judge=read();xi=read();//
		if(judge==1){
			ai=read();
			update(1,seg[xi],seg[xi],ai);
		}
		else
			if(judge==2){
				ai=read();
				update(1,seg[xi],seg[xi]+siz[xi]-1,ai);
			}
			else
				if(judge==3)
					printf("%lld\n",ask(1,xi));
	}
	return 0;
}