牛客 吉吉王国

链接：https://ac.nowcoder.com/acm/problem/210473
来源：牛客网

吉吉王国有nn个城市，其中11号城市就是吉吉王国的首都，并且吉吉王国有n-1n−1条道路，每条道路都有一个长度dd，你可以从任意一点uu到达任意一点vv。但是最近吉吉王国发生了动乱，除了首都外，每个只有一条道路连向的城市出现了反叛军。

这些城市的反叛军开始向吉吉王国的首都发起了进攻，如果让他们到达首都，那么吉吉国王就要换人了。吉吉国王现在需要快速切断一些道路，使得没有一个城市的反叛军可以到达首都。但是由于物资的限制，他能切断的道路的总长度和不能超过mm，并且要保证切断道路尽量快，因此切断的道路中最长的长度要尽可能小。

现在赶紧告诉吉吉国王切断的道路的最长长度在最小的时候是多少吧。
输入描述:
第一行两个整数n, mn,m。
接下来n-1n−1行每行三个整数u, v,du,v,d表示城市uu和城市vv有一条长度为dd的道路。
输出描述:
如果存在方案使得首都安全，输出最小的最长长度，否则输出-1。

思路：
和loves rinne很像，可以说是加强版
思路是因为有总长m和最长值最小的限制，因此可以使用二分答案，
二分最长值，用总长限制m来检验是否符合要求

#include<bits/stdc++.h>
using namespace std;
#define N 1010

const int INF=1e7+10;

int n,m;
int e[2*N],ne[2*N],w[N*2],h[N],idx=0;
int f[N];  //与loves ronne一样，以u为根的子树的最小值

void add(int a,int b,int c){
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u,int pre,int len){
    bool is_leaf=true;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(j==pre) continue;
        dfs(j,u,len);
        is_leaf=false;
        f[u]+=min(f[j],w[i]>len?INF:w[i]);
    }
    if(is_leaf) f[u]=INF;;
}

int main(){
    memset(h,-1,sizeof h);
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        int u,v,d;
        scanf("%d%d%d",&u,&v,&d);
        add(u,v,d),add(v,u,d);
    }
    
    int ans=0;
    
    int l=0,r=1e7+10;
    while(l<r){
        int mid=l+r>>1;
        memset(f,0,sizeof f);
        dfs(1,-1,mid);
        if(f[1]<=m) r=mid,ans=r;
        else l=mid+1;
    }
    
    if(!ans) puts("-1");
    else printf("%d\n",ans);
    
    return 0;
}