hdu 3586 tree dp

链接：http://acm.hdu.edu.cn/showproblem.php?pid=3586

题意：有n个士兵，他们之间通过n-1条线路通信，每条线路有个权值，构成树（哪里说了=。=...）。1为根节点，要使用动感光波切断根节点与所有叶子节点的联系，给出最大消耗（切断一条路就消耗相应的权值），求方案使切断的路的最大权值最小。

思路：最大值最小，和最小值最大这两个传说中经常用二分（不知道从哪里看到的。。），所以二分最大权值，判断在当前权值限制下能否成功，然后各种dp= =

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int n,m;
int l,r,mid;
const int maxn = 1100;
const int inf = 0x3f3f3f3f;
int node[maxn],top;
int dp[maxn];//dp[i]表示切断当前节点与他下面所有根节点所需要的最小消耗
struct Side{int v,w,next;}side[maxn];
void add_side(int u,int v,int w){
    side[top] = (Side){v,w,node[u]};
    node[u] = top++;
}
bool ok=1;
void dfs(int u){
    if(node[u]==-1){dp[u]=inf;return ;}//初始化为inf
    for(int i=node[u];i!=-1;i=side[i].next){
        int v=side[i].v;
        dfs(v);
        if(dp[v]>=inf&&side[i].w>mid)dp[u]=inf;
        else if(dp[v]>=inf)dp[u] += side[i].w;
        else if(side[i].w>mid)dp[u] += dp[v];
        else dp[u] += min(dp[v],side[i].w);
    }
}
int main(){
    while(~scanf("%d%d",&n,&m),n+m){
        memset(node,-1,sizeof(node));
        top = r = 0;
        for(int i=0;i<n-1;i++){
            int a,b,w;
            scanf("%d%d%d",&a,&b,&w);
            if(w>r)r=w;
            add_side(a,b,w);
        }
        l = 1;
        int mm = r;
        while(r>=l){
            memset(dp,0,sizeof(dp));
            mid = (l + r)>>1;
            dfs(1);
            if(dp[1]<=m)r = mid -1;
            else l = mid + 1;
        }
        if(l==mm+1)printf("-1\n");
        else printf("%d\n",l);
    }
    return 0;
}