19.10.01 acm E:Lowest Common Ancestor

题目描述

一棵有根树，对于每个点 $i$ ，求 $\sum_{j=1}^{i-1}w_{lca(i,j)}$

数据范围

$n \le 2 \times 10^5,1 \le w_i \le 10^4$

题解

我们可以考虑枚举 $lca$ 去更新答案

对于每个点 $x$ ，如果它成为两个点的 $lca$ ，那这两个点肯定来自不同的子树

那我们可以建立一个线段树表示这个区间的答案总和(?)，然后进行线段树合并即可

具体地话就是考虑合并到 $x,y$ 两个节点上，那 $x$ 的左儿子会对 $y$ 的右儿子造成贡献，同理 $y$ 的左儿子会对 $x$ 的右儿子造成贡献，所以我们再多维护一个区间有多少个数出现即可

代码

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5,M=N*50;
int a[N],TT,n,T[N],t,nx[N],hd[N],V[N];
int s[M],f[M],ls[M],rs[M],tg[M];
void add(int u,int v){
	nx[++t]=hd[u];V[hd[u]=t]=v;
}
#define mid ((l+r)>>1)
void insert(int &x,int l,int r,int v){
	s[x=++t]++;
	if (l==r) return;
	if (mid>=v) insert(ls[x],l,mid,v);
	else insert(rs[x],mid+1,r,v);
}
void down(int x){
	if (ls[x]) f[ls[x]]+=tg[x],tg[ls[x]]+=tg[x];
	if (rs[x]) f[rs[x]]+=tg[x],tg[rs[x]]+=tg[x];
	tg[x]=0;
}
int merge(int x,int y,int c,int d,int v){
	if (!x && !y) return 0;
	if (!x){f[y]+=d*v;tg[y]+=d*v;return y;}
	if (!y){f[x]+=c*v;tg[x]+=c*v;return x;}
	down(x);down(y);int z=++t;
	ls[z]=merge(ls[x],ls[y],c,d,v);
	rs[z]=merge(rs[x],rs[y],c+s[ls[y]],d+s[ls[x]],v);
	f[z]=f[ls[z]]+f[rs[z]];
	s[z]=s[ls[z]]+s[rs[z]];
	return z;
}
void dfs(int x,int fr){
	insert(T[x],1,n,x);
	for (int i=hd[x];i;i=nx[i])
		if (V[i]!=fr){
			dfs(V[i],x),
			T[x]=merge(T[x],T[V[i]],0,0,a[x]);
		}
}
void put(int x,int l,int r){
	if (l==r){
		if (l>1) printf("%d\n",f[x]);return;
	}
	down(x);put(ls[x],l,mid);put(rs[x],mid+1,r);
}
void work(){
	t=0;
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i]),hd[i]=0;
	for (int i=2,x;i<=n;i++)
		scanf("%d",&x),add(x,i);t=0;
	dfs(1,0);put(T[1],1,n);
	for (int i=1;i<=t;i++)
		ls[i]=rs[i]=tg[i]=f[i]=s[i]=0;
}
int main(){
	while(~scanf("%d",&n)) work();
	return 0;
}