[BZOJ5212][LCT][树形DP][树链剖分]ZJOI2018：历史

BZOJ5212
去年ZJOIDay1最可做的题吧

和SDOI2017树点涂色类似，但是难在修改和查询方式最大值

首先我们简化问题：给出树上每个点被access的次数，求最大的虚实边切换次数之和
一个点会被其子树和它自己影响，所以有一个比较普遍的结论，一个点的切换次数如果是它兄弟里面最大的，且小于总和的一半，那就轮流切换，否则总会有一些切换是无用的

所以就可以DP出一次的答案

考虑修改，修改会影响当前点到根这条的路径，如果一个点的贡献S大于它及其兄弟总和的一半，贡献加上去是不变的，那么就按照a进行轻重链剖分并维护全局ans即可

Code：

#include<bits/stdc++.h>
#define ll long long
#define int ll
using namespace std;
inline ll read(){
	ll res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=4e5+4;
int vis[N<<1],nxt[N<<1],head[N],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
ll ans,a[N];
int t[N];
namespace LCT{
	int ls[N],rs[N],fa[N];
	ll s[N],si[N];
	inline int isrs(int x){return x==rs[fa[x]];}
	inline bool isroot(int x){if(!fa[x]) return true; return x!=ls[fa[x]] && x!=rs[fa[x]];}
	inline void pushup(int x){s[x]=s[ls[x]]+s[rs[x]]+si[x]+a[x];}
	inline void rotate(int x){
		int y=fa[x],z=fa[y],b=(ls[y]==x?rs[x]:ls[x]);
		if(z && !isroot(y)) (ls[z]==y?ls[z]:rs[z])=x;
		fa[x]=z;fa[y]=x;b?fa[b]=y:0;
		if(ls[y]==x) rs[x]=y,ls[y]=b;
		else ls[x]=y,rs[y]=b;
		pushup(y);pushup(x);
	}
	inline void splay(int x){
		while(!isroot(x)){
			while(!isroot(fa[x])){
				if(isrs(x)==isrs(fa[x])) rotate(fa[x]);
				else rotate(x);
			}
			rotate(x);
		}
		pushup(x);
	}
}
using namespace LCT;
void dfs(int v){
	int hson=v;ll mx=a[v];
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(fa[v]==y) continue;
		fa[y]=v;dfs(y);
		si[v]+=s[y];
		if(mx<s[y]) mx=s[y],hson=y;
	}
	if((mx<<1) > (s[v]=si[v]+a[v])){
		ans+=(s[v]-mx)<<1;
		if(v!=hson) si[v]-=s[rs[v]=hson];
		else t[v]=1;
	}
	else t[v]=2,ans+=s[v]-1;
}
int n,m;
signed main(){
	n=read();m=read();
	for(int i=1;i<=n;i++) a[i]=read();
	for(int i=1;i<n;i++){
		int x=read(),y=read();
		add(x,y);add(y,x);
	}
	dfs(1);cout<<ans<<"\n";
	while(m--){
		int x=read();ll w=read();
		for(int y=0;x;x=fa[y=x]){
			splay(x);
			int S=s[x]-s[ls[x]];
			ans-=t[x]<2?(S-(t[x]?a[x]:s[rs[x]]))<<1:S-1;
			S+=w;s[x]+=w;(y?si:a)[x]+=w;
			if(s[y]<<1>S)si[x]+=s[rs[x]],si[x]-=s[rs[x]=y];
			if(s[rs[x]]<<1>S) t[x]=0,ans+=(S-s[rs[x]])<<1;
			else{
				if(rs[x]) si[x]+=s[rs[x]],rs[x]=0;
				if(a[x]<<1>S) t[x]=1,ans+=(S-a[x])<<1;
				else t[x]=2,ans+=S-1,rs[x]=0;
			}
		}
		cout<<ans<<"\n";
	}
	return 0;
}