【每日一题】Rinne Loves Edges

Rinne Loves Edges

https://ac.nowcoder.com/acm/problem/22598

solution

因为m=n,所以题目要求也就是要求断掉一些边,使得根与叶子节点不连通,用f[i]表示以i已经无法到达它子树中的所有叶子节点的最小花费。

转移的时候分两种情况,如果断开当前点与父亲相连的边,那么其子树中的边可以都不断开,花费为0.
如果不断开当前点与父亲相连的边,那么答案就是其儿子的f值之和。

树形dp即可。

code

/*
* @Author: wxyww
* @Date: 2020-04-02 20:47:38
* @Last Modified time: 2020-04-02 22:53:06
*/
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<bitset>
#include<cstring>
#include<algorithm>
#include<string>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
const int N = 100010;
ll read() {
    ll x=0,f=1;char c=getchar();
    while(c<'0'||c>'9') {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9') {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
struct node {
    int v,nxt,w;
}e[N << 1];
int head[N],ejs;
void add(int u,int v,int w) {
    e[++ejs].v = v;e[ejs].nxt = head[u];head[u] = ejs;e[ejs].w = w;
}
ll f[N];

void dfs(int u,int fa) {
    ll sum = 0;
    for(int i = head[u];i;i = e[i].nxt) {
        int v = e[i].v;
        if(v == fa) continue;
        f[v] = e[i].w;
        // printf("%d\n",e[i].w);
        dfs(v,u);
        sum += f[v];
    }
    if(sum)
    f[u] = min(f[u],sum);
}

int main() {
    memset(f,0x3f,sizeof(f));
    int n = read(),m = read(),S = read();
    for(int i = 1;i < n;++i) {
        int u = read(),v = read(),w = read();
        add(u,v,w);add(v,u,w);
    }
    dfs(S,0);
    cout<<f[S];
    return 0;
}
全部评论

相关推荐

10-30 22:18
已编辑
毛坦厂中学 C++
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
10-15 14:22
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务