P3258 [JLOI2014]松鼠的新家(树上差分板子)

树上差分！！！

这是一道模板题的稍微改编

首先，啥是树上差分

就和名字一样，书上的差分

倘若我们要给（u,v）的路径上加上k

则给u+k,v+k，然后lca（u,v）-k（因为事实上lca处就加了一个，可是这么干加了俩） father（lca（---））-k(这里及更高处根本没动过，消除影响)

所以这题就是lca+树上差分

但是和模板题的不同是，这里，每一点（除了一开始的和重点），即使起始点也是终点，而且最后一点不用放。

需要特别处理

      #include<iostream>
#include<cstdio>
using namespace std;
int head[2*500001];
int p;
struct b{
	int to;
	int ne;
} e[2*500001];
int fa[2*500001][30];
int dep[2*500001];
int n,m,s;
int logg[2*500001];
int ma[300005]; 
int cha[300005];
int ans[300005];
int cnt[300005];
int x,y;
void dfs(int now,int fat){
	dep[now]=dep[fat]+1;
	fa[now][0]=fat;
	for(int i=1;(1<<i)<=dep[now];++i){
		fa[now][i]=fa[fa[now][i-1]][i-1];
	}
	for(int i=head[now];i;i=e[i].ne){
		if(e[i].to!=fat)
		dfs(e[i].to,now);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]){
		swap(x,y);
	}
	while(dep[x]>dep[y])
	x=fa[x][logg[dep[x]-dep[y]]-1];
	if(x==y)
	return x;
	for(int k=logg[dep[x]]-1;k>=0;k--){
		if(fa[x][k]!=fa[y][k]){
			x=fa[x][k];
			y=fa[y][k];
		}
	}
	return fa[x][0];
}
void print(int x,int fa){
//	cout<<x<<endl;
	for(int i=head[x];i;i=e[i].ne){
//			cout<<e[i].to;
		if(e[i].to!=fa){
			print(e[i].to,x);
			ans[x]+=ans[e[i].to];
		}
	}
	ans[x]+=cha[x];
	return ;
}
void add(int f,int t){
	p++;
	e[p].to=t;
	e[p].ne=head[f];
	head[f]=p;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%d",&ma[i]);
		if(i!=1)
		cnt[ma[i]]++;
		}
	for(int i=1;i<=n-1;++i){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	for(int i=1;i<=n;++i)
		logg[i]=logg[i-1]+(1<<logg[i-1]==i);
	for(int i=1;i<n;++i){
		int o=lca(ma[i],ma[i+1]);
		int v=ma[i+1];
		cha[ma[i]]++;
		cha[v]++;
		cha[o]--;
		cha[fa[o][0]]--;
	}
	print(1,0);
	for(int i=1;i<=n;++i){
		ans[i]-=cnt[i];
		cout<<ans[i]<<endl;
	}
	return 0;
}