AcWing 352. 闇の連鎖 （树上差分---维护边权，LCA）

🔒 🔒 🔒
关于树上差分
（1）假设边权为删除该边之后还需要删除的边数，那么每在（x,y）上添加一条边，相当于将x到y 的路径上的边权全都加一。
（2）树上差分：以点值代表该点上面那条边的权值，那么要修改x,y路径上的所有边的权值，cha[x]+=w，cha[y]+=w，cha[ lca(x，y) ] -=w*2，之后dfs一遍求子树和即可

// lca
int dep[MX],dis[MX],f[MX][22];
//链式向前星
int n,m,tot,head[MX];
struct IN{int v,w,next;}edge[MX<<1];
void addedge(int u,int v,int w)
{
    edge[tot].v=v,edge[tot].w=w,edge[tot].next=head[u];
    head[u]=tot++;
}
void build()
{
    tot=1;
    memset(head,0,sizeof(head));
    for(int i=1; i<n; i++)
    {
        int u,v,w=1; scanf("%d %d",&u,&v);
        addedge(u,v,w);addedge(v,u,w);
    }
}
void lca_dfs(int u,int pre)
{
    for(int i=1; i<=20; i++)
    {
        f[u][i]=f[f[u][i-1]][i-1];
        if(!f[u][i])   break;
    }
    for(int i=head[u]; i; i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre)  continue;
        dep[v]=dep[u]+1,dis[v]=dis[u]+edge[i].w, f[v][0]=u;
        lca_dfs(v,u);
    }
}
void lca_init()
{
    dep[0]=-1,dep[1]=1,dis[1]=0;
    memset(dis,0,sizeof(dis));
    lca_dfs(1,-1);
}
int lca(int x,int y)
{
    if(dep[x]>dep[y])   swap(x,y);
    for(int i=20; i>=0; i--) if(dep[f[y][i]]>=dep[x]) y=f[y][i];
    for(int i=20; i>=0; i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    if(x!=y)  x=f[x][0];
    return x;
}
//树上差分
ll cha[MX];
void dfs(int x,int fx)//统计子树
{
    for(int i=head[x];i;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==fx) continue;
        dfs(v,x);
        cha[x]+=cha[v];
    }
}
signed main()
{
    cin>>n>>m;
    build();
    lca_init();
    rpp(i,m) 
    {
        int x,y;cin>>x>>y;
        ++cha[x],++cha[y],cha[lca(x,y)]-=2;
    }
    dfs(1,0);
    int ans=0;
    for(int i=2;i<=n;++i)
    {
        if(cha[i]==0) ans+=m;
        if(cha[i]==1) ans+=1;
    }
    cout<<ans<<endl;
    return 0;
}