AcWing 352. 闇の連鎖 //树上边差分

https://www.acwing.com/problem/content/description/354/

具体题解和树上边差分看这:https://www.cnblogs.com/gzh-red/archive/2019/07/14/11185914.html

这就是树上边差分的裸题。

树上边差分的实质就是:

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

在xi和yi打上标记ci,在lca(xi,yi)上打上标记-2*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 = 1e5+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 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];
    }
}

void Main(int avg){
    LL ans=0;int n,m;
    cin>>n>>m;
    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<=m;i++){
        int sa,sb;scanf("%d%d",&sa,&sb);
        addcf(sa,sb,1);
    }
    dfscf(1,-1);
    for(int i=2;i<=n;i++) {
        if(cf[i]==0) ans+=(LL)(m);
        if(cf[i]==1) ans++;
    }
    cout<<ans<<'\n';
}
int main(){
    int cas=1;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值