【BZOJ 4127】Abs 树链剖分

注意到后面加的每一个数都是>=0的，所以最多有n次从负数加到正数的情况，所以记录一个最大的负数，然后暴力处理从负加到正的情况就好了。

智障地方：1.树链剖分向上跳的时候比较的是h[a],h[b]而不是h[top[a]],h[top[b]]

2.update之前没有push_down()

好了，剩下的就是写代码了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<cmath>
#define maxn 100021
#define LL long long
#define ls u<<1,l,mid
#define rs u<<1|1,mid+1,r
using namespace std;
int n,m,w[maxn],head[maxn],h[maxn],son[maxn],cnt,cur[maxn];
int tot=1,size[maxn],top[maxn],f[maxn],nu[maxn];
struct edge{int v,next;}e[maxn*2];
void adde(int a,int b){e[tot].v=b,e[tot].next=head[a];head[a]=tot++;}
void dfs1(int u,int fa){
	h[u]=h[fa]+1,f[u]=fa,size[u]=1;
	for(int v,i=head[u];i;i=e[i].next){
		if((v=e[i].v)==fa)continue;
		dfs1(v,u);size[u]+=size[v];
		if(size[v]>size[son[u]])son[u]=v;
	}
}
void dfs2(int u,int fa,int tt){
	cur[nu[u]=++cnt]=u,top[u]=tt;
	if(son[u])dfs2(son[u],u,tt);
	for(int v,i=head[u];i;i=e[i].next){
		if((v=e[i].v)==fa||v==son[u])continue;
		dfs2(v,u,v);
	}
}
struct node{int Max,add,s1,s2,;LL sum1,sum2;}nod[maxn*4];
inline void push_up(int u){
	int ll=u<<1,rr=u<<1|1;
	nod[u].Max=max(nod[ll].Max,nod[rr].Max);
	nod[u].sum1=nod[ll].sum1+nod[rr].sum1;
	nod[u].sum2=nod[ll].sum2+nod[rr].sum2;
	nod[u].s1=nod[ll].s1+nod[rr].s1;nod[u].s2=nod[ll].s2+nod[rr].s2;
}
inline void push_down(int u){
	if(!nod[u].add)return ;
	int ll=u<<1,rr=u<<1|1,add=nod[u].add;
	if(nod[ll].Max!=-1e9)nod[ll].Max+=add;
	if(nod[rr].Max!=-1e9)nod[rr].Max+=add;
	nod[ll].sum2+=(LL)nod[ll].s2*add;
	nod[rr].sum2+=(LL)nod[rr].s2*add;
	nod[ll].sum1-=(LL)nod[ll].s1*add;
	nod[rr].sum1-=(LL)nod[rr].s1*add;
	nod[ll].add+=add,nod[rr].add+=add;
	nod[u].add=0;
}
void build(int u,int l,int r){
	if(l==r){
		nod[u].Max= w[cur[l]]<0 ? w[cur[l]] : -1e9;
		if(w[cur[l]]<0)nod[u].sum1=-w[cur[l]],nod[u].s1=1;
		else nod[u].sum2=w[cur[l]],nod[u].s2=1;
		return;
	}int mid=l+r>>1;
	build(ls),build(rs);
	push_up(u);
}
void update(int u,int l,int r,int x,int y,int add){
	int mid=l+r>>1;
	if(l==x&&r==y){
		if(nod[u].Max+add<0){
			nod[u].add+=add;
			if(nod[u].Max!=-1e9)nod[u].Max+=add;
			nod[u].sum2+=(LL)nod[u].s2*add;
			nod[u].sum1-=(LL)nod[u].s1*add;
			return;
		}else{
			if(l==r){
				nod[u].sum2=nod[u].Max+add;nod[u].sum1=0;
				nod[u].s2=1,nod[u].s1=0;
				nod[u].Max=-1e9,nod[u].add=0;
				return ;
			}else{
				push_down(u);
				update(ls,x,mid,add),update(rs,mid+1,y,add);
				push_up(u);return;
			}
		}
	}
	push_down(u);
	if(x>mid)update(rs,x,y,add);
	else if(y<=mid)update(ls,x,y,add);
	else update(ls,x,mid,add),update(rs,mid+1,y,add);
	push_up(u);
}
LL query(int u,int l,int r,int x,int y){
	if(x==l&&r==y)return nod[u].sum1+nod[u].sum2;
	int mid=l+r>>1;
	push_down(u);
	if(x>mid)return query(rs,x,y);
	else if(y<=mid)return query(ls,x,y);
	else return query(ls,x,mid)+query(rs,mid+1,y);
}

void change(int a,int b,int add){
	while(top[a]!=top[b]){
		if(h[top[a]]>h[top[b]])swap(a,b);
		update(1,1,n,nu[top[b]],nu[b],add);
		b=f[top[b]];
	}
	if(h[a]>h[b])swap(a,b);
	update(1,1,n,nu[a],nu[b],add);
}
void find(int a,int b){
	LL ans=0;
	while(top[a]!=top[b]){
		if(h[top[a]]>h[top[b]])swap(a,b);
		ans+=query(1,1,n,nu[top[b]],nu[b]);
		b=f[top[b]];
	}
	if(h[a]>h[b])swap(a,b);
	ans+=query(1,1,n,nu[a],nu[b]);
	printf("%lld\n",ans);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",w+i);
	for(int a,b,i=1;i<n;i++){
		scanf("%d%d",&a,&b);adde(a,b);adde(b,a);
	}dfs1(1,0);dfs2(1,1,1);
	build(1,1,n);int a,b,add,pos;
	while(m--){
		scanf("%d%d%d",&pos,&a,&b);
		if(pos==1){
			scanf("%d",&add);
			if(add)change(a,b,add);
		}else find(a,b);
	}
	return 0;
}