luogu P3258 [JLOI2014]松鼠的新家 //树上点差分

https://www.luogu.org/problem/P3258

树上点差分裸题

树上点差分的实质就是：

如果给m个(xi,yi)，要使得xi->yi的路径上的点加上一个权值ci，支持离线，O(nlogn)。

在xi和yi打上标记ci，在lca(xi,yi)打上标记-ci，在fa[lca(xi,yi)]打上标记-ci，最后离线统计。

#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const LL mod = 1e9+7;
const int MX = 3e5+50;
const int MXLOG = 25;

vector<int>g[MX];
int bei[MX][MXLOG],de[MX];
LL cf[MX];  //差分标记

void dfs(int v,int p){//所有都是向上走2^k步 所代表的含义
    for(auto i:g[v]) {
        if(i!=p){
            de[i]=de[v]+1;bei[i][0]=v;
            dfs(i,v);
        }
    }
}
void initLCA(int nn){
    bei[1][0]=-1;de[1]=0;dfs(1,-1);
    for(int k=0;k<MXLOG-1;k++){   //-1!!
        for(int v=1;v<=nn;v++){
            if(bei[v][k]<0)bei[v][k+1]=-1;
            else bei[v][k+1]=bei[bei[v][k]][k];
        }
    }
}
int LCA(int u,int v){
    if(de[u]>de[v]) swap(u,v);
    for(int k=0;k<MXLOG;k++){
        if((de[v]-de[u])>>k&1) v=bei[v][k];
    }
    if(u==v) return u;
    for(int k=MXLOG-1;k>=0;k--){
        if(bei[u][k]!=bei[v][k]){u=bei[u][k],v=bei[v][k];}
    }
    return bei[u][0];
}
/*
void addcf(int u,int v,LL c){ //(x->y)的路径上的边权值加c
    cf[u]+=c,cf[v]+=c,cf[LCA(u,v)]-=(c*2);
}
*/
void addcf(int u,int v,int c){
    int lca=LCA(u,v);
    cf[u]+=c,cf[v]+=c,cf[lca]-=c;
    if(bei[lca][0]!=-1) cf[bei[lca][0]]-=c;
}
void dfscf(int v,int p){  //标记转化成真实值后，cf[i]表示i到fa[i]的边的权值
    for(auto i:g[v]){
        if(i==p)continue;
        dfscf(i,v);
        cf[v]+=cf[i];
    }
}
int a[MX];
void Main(int avg){
    int n,m;
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<n;i++){
        int sa,sb;scanf("%d%d",&sa,&sb);g[sa].push_back(sb);g[sb].push_back(sa);
    }
    initLCA(n);
    for(int i=1;i<n;i++) addcf(a[i],a[i+1],1);
    dfscf(1,-1);
    for(int i=2;i<=n;i++) cf[a[i]]--;
    for(int i=1;i<=n;i++)cout<<cf[i]<<'\n';
}
int main(){
    int cas=1;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}