[JLOI2014]松鼠的新家

传送门

树上差分

统计一条路径上所有点的经过次数,我们通过统计边的经过次数来统计点的经过次数。

差分统计的对象变为边。每个点的子树的点权和表示这个点到父节点相连边的经过次数，差分时修改LCA的权值$-2$。

#include<bits/stdc++.h>
#define il inline
using namespace std;
const int N=3e5+5;
#define getchar()(p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
il int read(){
	int x=0,f=1;char c=getchar();
	for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
	for(;isdigit(c);c=getchar()) x=(x+(x<<2)<<1)+c-48;
	return x*f;
}
char sr[1<<21],z[20];int C=-1,Z;
il void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
il void print(int x){
	if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
	while(z[++Z]=x%10+48,x/=10);
	while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
int n,k,t,h[N],f[N][25],d[N],s[N],cnt[N];
struct Edge{int to,nxt,val;}a[N<<1];
il void add(int x,int y){a[++k].to=y,a[k].nxt=h[x],h[x]=k;}
il void dfs(int x,int fa){
	d[x]=d[fa]+1;
	for(int i=h[x];i;i=a[i].nxt){
		int y=a[i].to;
		if(y==fa) continue;
		f[a[i].to][0]=x;
		for(int j=1;j<=t;++j)
		f[y][j]=f[f[y][j-1]][j-1];
		dfs(a[i].to,x);
	}
}
il int lca(int x,int y){
	if(d[x]>d[y]) swap(x,y);
	for(int i=t;i>=0;--i)
	if(d[f[y][i]]>=d[x]) y=f[y][i];
	if(x==y) return x;
	for(int i=t;i>=0;--i)
	if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}
il int dfs2(int x,int fa){
	int ans=cnt[x];
	for(int i=h[x];i;i=a[i].nxt)
	if(a[i].to!=fa) ans+=a[i].val=dfs2(a[i].to,x);
	for(int i=h[x];i;i=a[i].nxt)
	if(a[i].to==fa) a[i].val=ans,i=0;
	return ans;
}
int main(){
	n=read(),t=log2(n)+1;
	for(int i=1;i<=n;++i) s[i]=read();
	for(int i=1;i<n;++i){
		int x=read(),y=read();
		add(x,y),add(y,x);
	}
	f[s[1]][0]=s[1];
	dfs(s[1],0);
	for(int i=1;i<n;++i) ++cnt[s[i]],++cnt[s[i+1]],cnt[lca(s[i],s[i+1])]-=2;
	dfs2(s[1],0);
	for(int i=1;i<=n;++i){
		int ans=0;
		for(int j=h[i];j;j=a[j].nxt) ans+=a[j].val;
		if(i==s[n]) --ans;print((ans+1)>>1);
	}Ot();return 0;
}