BZOJ3631 LCA

根本不用树链剖分

直接类似数组加个标记dfs就行

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int A[N],ver[N*2],Next[N*2],head[N],d[N],DP[N],f[N][20],t,tot;
queue<int> q;
void add(int x,int y){
	ver[++tot]=y;Next[tot]=head[x];head[x]=tot;
}
void bfs(){
	q.push(1),d[1]=1;
	while(q.size()){
		int x=q.front();q.pop();
		for(int i=head[x];i;i=Next[i]){
			int y=ver[i];if(d[y])continue;
			d[y]=d[x]+1;f[y][0]=x;
			for(int j=1;j<=t;++j)f[y][j]=f[f[y][j-1]][j-1];
			q.push(y);
		}
	}
}
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];
}
void dfs(int x,int y){
	for(int i=head[x];i;i=Next[i]){
		if(ver[i]==y)continue;
		dfs(ver[i],x);
		DP[x]+=DP[ver[i]];
	}
}
int main(){
	int n,x,y;
	cin>>n;
	t=int(log(n)/log(2))+1;
	for(int i=1;i<=n;++i)scanf("%d",&A[i]);
	for(int i=1;i<n;++i)scanf("%d%d",&x,&y),add(x,y),add(y,x);
	bfs();
	for(int i=1;i<n;++i){
		int x=lca(A[i],A[i+1]);
		DP[f[x][0]]--;
		DP[x]--;
		DP[f[A[i]][0]]++;
		DP[A[i+1]]++;
	}
	dfs(1,0);
	DP[A[1]]++;
	DP[A[n]]--;
	for(int i=1;i<=n;++i)printf("%d\n",DP[i]);
}