99网
您的当前位置:首页牛客小白月赛11 F. Rinne Loves Edges

牛客小白月赛11 F. Rinne Loves Edges

来源:99网

链接

题意

边权树形图上删去一些边使所有的叶子节点都无法到达根节点,求删去边的权值和的最小值

思路

树形 DP

让子树上的叶子节点和根节点不连通,显然只有两种情况:让叶子节点断开或让子树断开

f [ x ] f[x] f[x] 为让 x x x 的子树上的叶子节点与根节点断开的最小代价, d i s t [ x ] dist[x] dist[x] x x x 与其父亲节点之间的距离:

f [ x ] = m i n ( s u m ( f [ y ] ) , d i s t [ x ] ) ( y ∈ s o n ( x ) ) f[x]=min(sum(f[y]),dist[x])(y \in son(x)) f[x]=min(sum(f[y]),dist[x])(yson(x))

代码

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
typedef long long ll;
typedef pair<int,int> pii;
const int N=100010;
int n,m,s;
vector<pii>edge[N];
ll f[N];
void dfs(int u,int fa,ll w) {
    ll tot=0;
    for(auto x:edge[u]) {
        if(x.fi==fa) continue;
        dfs(x.fi,u,x.se);
        tot+=f[x.fi];
    }
    if(tot) f[u]=min(w,tot);
    else f[u]=w;
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m>>s;
    for(int i=1;i<=m;i++) {
        int u,v,w;
        cin>>u>>v>>w;
        edge[u].pb({v,w});
        edge[v].pb({u,w});
    }
    memset(f,0x3f,sizeof f);
    dfs(s,0,1e18);
    cout<<f[s]<<endl;
    return 0;
}

因篇幅问题不能全部显示,请点此查看更多更全内容