luogu P3258 [JLOI2014]松鼠的新家

题面传送门
只有一次查询，想到差分。
那么我们怎么把差分运用到树上呢?
自然要在\(lca\)上下文章了。
可是不能直接在\(lca\)上下，得在\(fa_{lca}\)上下，即\(f_u++，f_v++，f_{lca(x,y)}--，f_{fa_{lca(x,y)}}--\)
注意不是每个点都出现，我因为这个调了一天。
代码实现:

#include<cstdio>
#include<cstring>
using namespace std;
int a[300039],fa[300039][20],n,m,k,x,y,lg[300039],d[300039],f[300039],lcas,ls;
struct yyy{int to,z;};
struct ljb{
	yyy f[600039];
	int head,h[300039];
	inline void add(int x,int y){
		f[++head]=(yyy){y,h[x]};
		h[x]=head;
	}
}s;
inline void dfs1(int x,int last){
	fa[x][0]=last;
	d[x]=d[last]+1;
	for(int i=1;i<=lg[d[x]];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
	int cur=s.h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=s.f[cur];
		if(tmp.to!=last) dfs1(tmp.to,x);
		cur=tmp.z;
	}
}
inline void swap(int &x,int &y){x^=y,y^=x,x^=y;}
inline int lca(int x,int y){
	if(d[x]<d[y]) swap(x,y);
	while(d[x]>d[y]) x=fa[x][lg[d[x]-d[y]]-1];
	if(x==y) return x;
	for(int i=lg[d[x]]-1;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
inline void dfs2(int x){
	int cur=s.h[x];
	yyy tmp;
	//printf("%d\n",x);
	while(cur!=-1){
		tmp=s.f[cur];
		if(tmp.to!=fa[x][0]) dfs2(tmp.to),f[x]+=f[tmp.to];
		cur=tmp.z;
	}
}
int main(){
//	freopen("1.in","r",stdin);
//	freopen("1.out","w",stdout);
	register int i,j;
	scanf("%d",&n);
	for(i=1;i<=n;i++) s.h[i]=-1;
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
	for(i=1;i<=n;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(i=1;i<n;i++) scanf("%d%d",&x,&y),s.add(x,y),s.add(y,x);
	dfs1(1,0);
	for(i=2;i<=n;i++){
		lcas=lca(a[i],a[i-1]);
		f[a[i]]++;
		f[a[i-1]]++;
		f[lcas]--;
		f[fa[lcas][0]]--;
	}
	dfs2(1);
	for(i=2;i<=n;i++) f[a[i]]--;
	for(i=1;i<=n;i++) printf("%d\n",f[i]);
}