[LOJ2135][点分树]ZJOI2015：幻想乡战略游戏

LOJ2135

显然点分树维护，维护一个点在点分树上的子树权值和，子树到它的代价和，子树到它父亲的代价和，用加减消掉询问时当前子树的影响，这些都是点分树的套路
修改暴力爬树
然后询问先从点分树根节点开始，算一次根节点的答案，然后和各个儿子的答案比较一下，如果某个儿子比它优，就进这个儿子找
这样询问的正确性：如果这个儿子里面没有答案，那这个儿子肯定比根节点差
如果这个儿子里面有答案，那这个儿子肯定比根节点优，所以一个点就只可能递归到一个儿子里去，复杂度$log$，加上爬树一次询问是$lo{g}^2$

Code：

#include<bits/stdc++.h>
#define ll long long
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=1e5+5,INF=1e9;
int vis[N<<1],head[N],nxt[N<<1],tot=0;
ll c[N<<1];
inline void add(int x,int y,ll z){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;c[tot]=z;}
int Vis[N<<1],Head[N],Nxt[N<<1],Tot=0;
ll C[N<<1];
inline void Add(int x,int y,ll z){Vis[++Tot]=y;Nxt[Tot]=Head[x];Head[x]=Tot;C[Tot]=z;}
int dp[N],fa[N][20],siz[N],pt[N];
int sum,rt;
void getroot(int v,int f){
	siz[v]=1;dp[v]=0;
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(y==f || pt[y]) continue;
		getroot(y,v);
		siz[v]+=siz[y];
		dp[v]=max(dp[v],siz[y]);
	}
	dp[v]=max(dp[v],sum-siz[v]);
	if(dp[v]<dp[rt]) rt=v;
}
int d[N];
ll dis[N][20];
void getship(int v,int anc,int f,int dd){
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(pt[y] || y==f) continue;
		fa[y][++d[y]]=anc;
		dis[y][d[y]]=dd+c[i];
		getship(y,anc,v,dd+c[i]);
	}
}
void buildtree(int v){
	pt[v]=1;getship(v,v,0,0);
	int all=sum;
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(pt[y]) continue;
		sum=siz[y];if(sum>siz[v]) sum=all-siz[v];
		rt=0;getroot(y,v);Add(v,rt,y);buildtree(rt);
	}
}
ll sumv[N],sumfa[N],sumnow[N];
inline void modify(int v,ll val){
	sumv[v]+=val;
	for(int i=d[v];i;i--){
		ll tmp=dis[v][i]*val;
		sumnow[fa[v][i]]+=tmp;
		sumfa[fa[v][i+1]]+=tmp;
		sumv[fa[v][i]]+=val;
	}
}
inline ll calc(int v){
	ll res=sumnow[v];
	for(int i=d[v];i;i--){
		ll tmp=dis[v][i];
		res+=sumnow[fa[v][i]]-sumfa[fa[v][i+1]];
		res+=tmp*(sumv[fa[v][i]]-sumv[fa[v][i+1]]);
	}
	return res;
}
ll query(int v){
	ll ans=calc(v);
	for(int i=Head[v];i;i=Nxt[i]){
		int y=C[i],z=Vis[i];
		ll tmp=calc(y);
		if(tmp<ans) return query(z);
	}
	return ans;
}
int main(){
	int n=read(),q=read();
	for(int i=1;i<n;i++){
		int x=read(),y=read();ll z=read();
		add(x,y,z);add(y,x,z);
	}
	dp[0]=n+1;rt=0;sum=n;getroot(1,0);int st=rt;buildtree(rt);
	for(int i=1;i<=n;i++) fa[i][d[i]+1]=i;
	while(q--){
		int x=read();ll y=read();modify(x,y);
		cout<<query(st)<<"\n";
	}
	return 0;
}