洛谷P3676 小清新数据结构题 动态点分治

题目分析

设整棵树的点权和为$W$，$s_i$表示以$i$为根的子树的点权和，则无论哪个点为根，${\sum }_{i=1}^n{s}_i(W-s_i)=P$都是个定值。

这是因为可以看做当两个点$i,j$分别在以某个点为根的子树内外时，$P$加上$v_i{v}_j$的贡献，也就是说$P={\sum }_{i=1}^n{\sum }_{j=1}^{n}dist(i,j)v_i{v}_j$（维护$P$也是用这个式子维护）。

于是${\sum }_{i=1}^n{s}_i^2=P-{\sum }_{i=1}^n{s}_{i}W$

那么我们只要求以$x$为根时的${\sum }_{i=1}^n{s}_i$了。把统计答案的方法改成，每个点$i$往上跳祖先，跳到一个点，就产生$v_i$的贡献，那么答案就是${\sum }_{i=1}^n(dist(i,x)+1)v_i={\sum }_{i=1}^{n}dist(i,x)v_i+W$。

用动态点分治，可以做到询问每个点$x$意义下的${\sum }_{i=1}^{n}dist(i,x)v_i$。于是就能维护以$1$为根意义下的$P$，还有询问了。

代码

树剖求LCA这么麻烦，倍增不就可以了吗，复杂度都是一样的，我才不写树剖呢(￣_,￣ )

树剖 0.4s，倍增 1.6s

真香。

#include<bits/stdc++.h>
using namespace std;
#define RI register int
int read() {
	int q=0,w=1;char ch=' ';
	while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
	if(ch=='-') w=-1,ch=getchar();
	while(ch>='0'&&ch<='9') q=q*10+ch-'0',ch=getchar();
	return q*w;
}
typedef long long LL;
const int N=200005,inf=0x3f3f3f3f;
int n,Q,tot,mi,rt;LL W,P;
int h[N],ne[N<<1],to[N<<1],sz[N],vis[N],fa[N],dep[N],f[N][18];
LL v[N],pre_sum[N],dsum[N],dsum_fa[N],sum[N];

void add(int x,int y) {to[++tot]=y,ne[tot]=h[x],h[x]=tot;}
void pre_dfs(int x,int las) {
	dep[x]=dep[las]+1,f[x][0]=las,pre_sum[x]=v[x];
	for(RI i=1;i<=17;++i) f[x][i]=f[f[x][i-1]][i-1];
	for(RI i=h[x];i;i=ne[i])
		if(to[i]!=las) pre_dfs(to[i],x),pre_sum[x]+=pre_sum[to[i]];
	P+=pre_sum[x]*(W-pre_sum[x]);
}
int lca(int x,int y) {
	if(dep[x]<dep[y]) swap(x,y);
	for(RI i=17;i>=0;--i) if(dep[f[x][i]]>=dep[y]) x=f[x][i];
	if(x==y) return x;
	for(RI i=17;i>=0;--i) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}
LL dist(int x,int y) {return dep[x]+dep[y]-2LL*dep[lca(x,y)];}

void getrt(int x,int las,int SZ) {
	sz[x]=1;int bl=0;
	for(RI i=h[x];i;i=ne[i])
		if(!vis[to[i]]&&to[i]!=las)
			getrt(to[i],x,SZ),sz[x]+=sz[to[i]],bl=max(bl,sz[to[i]]);
	bl=max(bl,SZ-sz[x]);
	if(bl<mi) mi=bl,rt=x;
}
void work(int x) {
	vis[x]=1;
	for(RI i=h[x];i;i=ne[i])
		if(!vis[to[i]]) mi=inf,getrt(to[i],x,sz[to[i]]),fa[rt]=x,work(rt);
}

void chan(int x,LL val) {
	int y=x,las=0;
	while(y) {
		LL kd=dist(x,y);dsum[y]+=kd*val,sum[y]+=val;
		if(las) dsum_fa[las]+=kd*val;
		las=y,y=fa[y];
	}
}
LL query(int x) {
	int y=x,las=0;LL re=0;
	while(y) {
		re+=(dsum[y]-dsum_fa[las])+(sum[y]-sum[las])*dist(y,x);
		las=y,y=fa[y];
	}
	return re;
}

int main()
{
	int x,y,op;
	n=read(),Q=read();
	for(RI i=1;i<n;++i) x=read(),y=read(),add(x,y),add(y,x);
	for(RI i=1;i<=n;++i) v[i]=read(),W+=v[i];
	pre_dfs(1,0),mi=inf,getrt(1,0,n),work(rt);
	for(RI i=1;i<=n;++i) chan(i,v[i]);
	while(Q--) {
		op=read(),x=read();
		if(op==1) {
			y=read(),W+=y-v[x],chan(x,y-v[x]);
			P+=(y-v[x])*query(x),v[x]=y;
		}
		else printf("%lld\n",(query(x)+W)*W-P);
	}
	return 0;
}