bzoj3631[JLOI2014 松鼠的新家 倍增lca+差分

裸的树上差分+倍增lca

每次从起点到终点左闭右开，这就有一个小技巧，要找到右端点向左端点走的第一步，然后差分就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define N 300005
using namespace std;
int fa[N][20],dep[N],f[N],g[N],n,l[N];
int e=1,head[N];
struct edge{
	int u,v,next;
}ed[2*N];
void add(int u,int v){
	ed[e].u=u;ed[e].v=v;
	ed[e].next=head[u];
	head[u]=e++;
}
void dfs(int x){
	for(int i=1;(1<<i)<=dep[x];i++)
		fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int i=head[x];i;i=ed[i].next){
		int v=ed[i].v;
		if(v==fa[x][0])continue;
		fa[v][0]=x;dep[v]=dep[x]+1;
		dfs(v);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=19;i>=0;i--)
		if(dep[fa[x][i]]>=dep[y])
			x=fa[x][i];
	if(x==y)return x;
	for(int i=19;i>=0;i--)
		if(fa[x][i]!=fa[y][i])
			x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int find(int x,int y){
	int d=dep[x]-y;
	for(int i=19;~i;i--)
		if(dep[fa[x][i]]>=d)
			x=fa[x][i];
	return x;
}
void update(int x,int y){
	f[x]++;f[y]++;
	int z=lca(x,y);
	f[z]--;f[fa[z][0]]--;
}
void work(int x,int y){
	if(dep[x]<=dep[y])update(x,fa[y][0]);
	else{
		int d=dep[x]-dep[y];
		int z=find(x,d-1);
		if(fa[z][0]==y)update(x,z);
		else update(x,fa[y][0]);
	}
}
void dfs1(int x){
	g[x]=f[x];
	for(int i=head[x];i;i=ed[i].next){
		if(ed[i].v==fa[x][0])continue;
		dfs1(ed[i].v);
		g[x]+=g[ed[i].v];
	}
}
int main(){
	scanf("%d",&n); int u,v;
	for(int i=1;i<=n;i++)scanf("%d",&l[i]);
	for(int i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	dep[1]=1; dfs(1);
	for(int i=1;i<n;i++)
		work(l[i],l[i+1]);
	dfs1(1);
	for(int j=1;j<=n;j++)
		printf("%d\n",g[j]);
	return 0;
}