模板-树上边差分

题目链接:松鼠采松果

将边权分给边相邻的较深的点,化为树上点差分。

模板:

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 2e5+5;
int n,m,q;
struct node{
    int x,y;
};
vector<node>g[N];
int d[N],fa[N][35],dep[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,w]: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(y==x)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,w]:g[x]){
        if(y==p)continue;
        dfs1(y,x);
        d[x]+=d[y];
    }
}

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

void solve(){
	cin>>n>>m>>q;
    for(int i=1;i<n;i++){
        int x,y,w;cin>>x>>y>>w;
        g[x].push_back({y,w});
        g[y].push_back({x,w});
    }

    dfs(1,0);

    for(int i=1;i<=m;i++){
        int x,y,w;cin>>x>>y>>w;
        d[x]+=w;
        d[y]+=w;
        d[lca(x,y)]-=2LL*w;
    }

    dfs1(1,0);//求原数组

    dfs2(1,0);//求前缀和

    while(q--){
        int x,y;cin>>x>>y;
        cout<<d[x]+d[y]-2LL*d[lca(x,y)]<<"\n";
    }

}

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

    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值