根本不用树链剖分
直接类似数组加个标记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]);
}