模板-树上点差分

题目链接：松鼠的新家

图解：

模板：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 3e5+5;
int n;
vector<int>g[N];
int d[N],fa[N][35],dep[N];
int a[N];

void dfs(int x,int p){
    dep[x]=dep[p]+1;
    fa[x][0]=p;

    for(int i=1;i<=31;i++){
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }

    for(const auto &y:g[x]){
        if(y==p)continue;
        dfs(y,x);
    }
    return;
}

int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);

    for(int i=31;i>=0;i--){
        if(dep[fa[x][i]]>=dep[y]){
            x=fa[x][i];
        }
    }

    if(x==y)return x;

    for(int i=31;i>=0;i--){
        if(fa[x][i]!=fa[y][i]){
            x=fa[x][i];
            y=fa[y][i];
        }
    }

    return fa[x][0];

}

void dfs1(int x,int p){
    for(const auto &y:g[x]){
        if(y==p)continue;
        dfs1(y,x);
        d[x]+=d[y];
    }
    return;
}

void solve(){
	cin>>n;
    memset(d,0,sizeof(d));
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<n;i++){
        int x,y;cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }

    dfs(1,0);

    for(int i=2;i<=n;i++){
        int u=a[i-1];
        int v=a[i];
        int c=lca(u,v);
        d[u]++;
        d[v]++;
        d[c]--;
        d[fa[c][0]]--;
        //cout<<u<<" "<<v<<" "<<c<<"\n";
    }

    dfs1(1,0);//差分前缀和

    for(int i=2;i<=n;i++){
        d[a[i]]--;//因为我们把既作为起点又做为终点的点算了两次，现在把这些点的权值减一即可 
    }

    for(int i=1;i<=n;i++){
        cout<<d[i]<<"\n";
    }

}

signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t=1;
    while(t--){
        solve();
    }

    return 0;
}