「BZOJ1036」[ZJOI2008] 树的统计Count

基本熟练运用树剖
但是还是有点离不开标程

传送门
黄学长题解
顺便吐槽BZOJ进不去了

Description
一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

code
不过似乎不需要lazy_tag

#include<bits/stdc++.h>
#define re register
#define in Read()
int in{
	int i=0,f=1;char ch=getchar();
	while(ch!='-'&&!isdigit(ch))ch=getchar();
	if(ch=='-')ch=getchar(),f=-1;
	while(isdigit(ch))i=(i<<1)+(i<<3)+ch-48,ch=getchar();
	return i*f;
}
const int NNN=3e4+10;
const int inf=2147483647;
int n,w[NNN],wt[NNN];

using std::swap;
using std::max;
using std::vector;
vector<int>G[NNN];

int father[NNN],son[NNN],top[NNN],dep[NNN],size[NNN],num[NNN],cnt;

namespace SGT{
	
	const int NNN=::NNN<<2;
	int tsum[NNN],tmax[NNN],lazy[NNN];
	
	inline void push_up(int p){
		tsum[p]=tsum[p<<1]+tsum[p<<1|1];
		tmax[p]=max(tmax[p<<1],tmax[p<<1|1]);
	}
	
	inline void push_down(int p,int len){
		lazy[p<<1]+=lazy[p];
		lazy[p<<1|1]+=lazy[p];
		tsum[p<<1]+=lazy[p]*(len-(len>>1));
		tsum[p<<1|1]+=lazy[p]*(len>>1);
		lazy[p]=0;
	}
	
	inline void build(int p,int l,int r){
		if(!(l^r)){
			tsum[p]=tmax[p]=wt[l];
			return ;
		}
		int mid=(l+r)>>1;
		build(p<<1,l,mid);
		build(p<<1|1,mid+1,r);
		push_up(p);
	}
	
	inline void update(int p,int l,int r,int x,int t){
		if(!(l^r)){
			lazy[p]+=t-tsum[p];
			tsum[p]=tmax[p]=t;
			return ;
		}
		int mid=(l+r)>>1;
		if(lazy[p])push_down(p,r-l+1);
		if(x<=mid)update(p<<1,l,mid,x,t);
		else update(p<<1|1,mid+1,r,x,t);
		push_up(p);
	}
	
	inline int qmax(int p,int l,int r,int L,int R){
		if(L<=l&&r<=R){
			return tmax[p];
		}
		int res=-inf;
		int mid=(l+r)>>1;
		if(mid>=L)res=max(res,qmax(p<<1,l,mid,L,R));
		if(mid<R)res=max(res,qmax(p<<1|1,mid+1,r,L,R));
		return res;
	}
	
	inline int qsum(int p,int l,int r,int L,int R){
		if(L<=l&&r<=R){
			return tsum[p];
		}
		int res=0;
		int mid=(l+r)>>1;
		if(lazy[p])push_down(p,r-l+1);
		if(mid>=L)res+=qsum(p<<1,l,mid,L,R);
		if(mid<R)res+=qsum(p<<1|1,mid+1,r,L,R);
		return res;
	}
	
}

inline void pre_DFS(int u,int fa,int depth){
	dep[u]=depth;
	father[u]=fa;
	size[u]=1;
	int heavy_son=0;
	for(re int v:G[u])
		if(v^fa){
			pre_DFS(v,u,depth+1);
			size[u]+=size[v];
			if(heavy_son<size[v])
				heavy_son=size[v],son[u]=v;
		}
}

inline void DFS(int u,int roof){
	num[u]=++cnt;
	top[u]=roof;
	wt[cnt]=w[u];
	if(!son[u])return ;
	DFS(son[u],roof);
	for(re int v:G[u])
		if((v^son[u])&&(v^father[u]))
			DFS(v,v);
}

inline void pre_work(){
	pre_DFS(1,0,1);
	DFS(1,1);
	SGT::build(1,1,n);
}

inline void Change(int u,int t){
	SGT::update(1,1,n,num[u],t);
}

inline int query_max(int u,int v){
	int res=-inf;
	while(top[u]^top[v]){
		if(dep[top[u]]<dep[top[v]])swap(u,v);
		res=max(res,SGT::qmax(1,1,n,num[top[u]],num[u]));
		u=father[top[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	res=max(res,SGT::qmax(1,1,n,num[u],num[v]));
	return res;
}

inline int query_sum(int u,int v){
	int res=0;
	while(top[u]^top[v]){
		if(dep[top[u]]<dep[top[v]])swap(u,v);
		res+=SGT::qsum(1,1,n,num[top[u]],num[u]);
		u=father[top[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	res+=SGT::qsum(1,1,n,num[u],num[v]);
	return res;
}

int main(){
	n=in;
	for(re int i=1;i< n;++i){
		int u=in,v=in;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	
	for(re int i=1;i<=n;++i)w[i]=in;
	
	pre_work();
	
	using std::string;
	using std::cin;
	int q=in;
	for(re int i=1;i<=q;++i){
		string Query;
		cin>>Query;
		if(Query=="CHANGE"){int u=in,t=in;Change(u,t);}
		else if(Query=="QMAX"){int u=in,v=in;printf("%d\n",query_max(u,v));}
		else if(Query=="QSUM"){int u=in,v=in;printf("%d\n",query_sum(u,v));}
	}
	return 0;
}