[洛谷P3258] [JLOI2014]松鼠的新家（树链剖分）

直接树链剖分，依次把从ai到ai+1的路径上的点都+1。最后输出时除了a1其他都-1，因为从上一个点到这个点和从这个点出发总共加了两次。

#include<cstdio>
#include<cstring>
const int N=300010;
struct edge{
	int y,next;
}data[N*2];
struct node{
	int a,b,s;
}tree[N*4];
int n,num,num1,a1[N],b1[N],h[N],son[N],siz[N],dep[N],faz[N],top[N],tid[N],rnk[N];
inline int read(){
	int x=0,f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return f?-x:x;
}
void dfs1(int u,int fa,int step){
	faz[u]=fa;son[u]=-1;siz[u]=1;dep[u]=step;
	for(int i=h[u];i!=-1;i=data[i].next){
		int v=data[i].y;
		if(v!=fa){
			dfs1(v,u,step+1);
			if(son[u]==-1||siz[son[u]]<siz[v])son[u]=v;
			siz[u]+=siz[v];
		}
	}
}
void dfs2(int u,int fa,int t){
	top[u]=t;tid[u]=++num1;rnk[num1]=u;
	if(son[u]==-1)return;
	dfs2(son[u],u,t);
	for(int i=h[u];i!=-1;i=data[i].next){
		int v=data[i].y;
		if(v!=fa&&v!=son[u])dfs2(v,u,v);
	}
}
void build(int p,int a,int b){
	tree[p].a=a;tree[p].b=b;tree[p].s=0;
	if(a==b)return;
	int mid=(a+b)>>1;
	build(p<<1,a,mid);build(p<<1|1,mid+1,b);
}
inline void new1(int p,int f){
	tree[p].s+=f;
}
inline void pushdown(int p){
	if(tree[p].s){new1(p<<1,tree[p].s);new1(p<<1|1,tree[p].s);tree[p].s=0;}
}
void add(int p,int a,int b){
	if(a<=tree[p].a&&tree[p].b<=b){new1(p,1);return;}
	pushdown(p);
	int mid=(tree[p].a+tree[p].b)>>1;
	if(a<=mid)add(p<<1,a,b);
	if(mid<b)add(p<<1|1,a,b);
}
void lca_add(int x,int y){
	int fx=top[x],fy=top[y];
	while(fx!=fy){
		if(dep[fx]<dep[fy]){
			add(1,tid[fy],tid[y]);
			y=faz[fy];fy=top[y];
		}else{
			add(1,tid[fx],tid[x]);
			x=faz[fx];fx=top[x];
		}
	}
	if(dep[x]<dep[y])add(1,tid[x],tid[y]);
	else add(1,tid[y],tid[x]);
}
void query(int p){
	if(tree[p].a==tree[p].b){
		if(rnk[tree[p].a]==a1[1])b1[rnk[tree[p].a]]=tree[p].s;
		else b1[rnk[tree[p].a]]=tree[p].s-1;
		return;
	}
	pushdown(p);
	query(p<<1);query(p<<1|1);
}
int main(){
	n=read();
	for(int i=1;i<=n;i++)a1[i]=read();
	memset(h,-1,sizeof h);num=num1=0;
	for(int x,y,i=1;i<n;i++){
		x=read();y=read();
		data[++num].y=y;data[num].next=h[x];h[x]=num;
		data[++num].y=x;data[num].next=h[y];h[y]=num;
	}
	dfs1(1,0,1);dfs2(1,0,1);
	build(1,1,n);
	for(int i=1;i<n;i++)lca_add(a1[i],a1[i+1]);
	query(1);
	for(int i=1;i<=n;i++)printf("%d\n",b1[i]);
	return 0;
}