HDU 3586 Information Disturbing

题意：N个人组成一棵树，编号1-N，其中编号为1的人为总司令，叶节点为前线士兵。点与点之间的边有一定的权值，代表砍掉这条边所要花费的代价。题目要求砍掉某些边使得所有的叶节点都无法与根节点联通，而且在满足砍掉的所有的边的和<=m的情况下的最小限制，这里的限制是指砍掉的边中的权值最大的数。

分析：树形DP+二分。二分限制limit，设sum[i]为砍掉以节点i为根节点的所有叶节点的费用总和，则有

if(tree[u][i].w>mid) sum[u]+=sum[v];
else                       sum[u]+=Min(sum[v],tree[u][i].w); v为节点u的子节点。

Code：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <vector>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
//#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;

const int maxn=1005;
const int inf=1000005;
struct edge {
    int v,w;
};
vector<edge>tree[maxn];
int sum[maxn];
int n,m,mid;

void init() {
    for(int i=1;i<=n;i++) tree[i].clear();
}

void add_edge(int u,int v,int w) {
    edge e;
    e.v=v;
    e.w=w;
    tree[u].push_back(e);
}

void dfs(int u,int fa) {
    int sz=tree[u].size();
    if(sz<=1 && tree[u][0].v==fa) {//根节点
        sum[u]=inf;
        return ;
    }
    sum[u]=0;
    for(int i=0;i<sz;i++) {
        int v=tree[u][i].v;
        if(v==fa) continue;
        dfs(v,u);
        if(tree[u][i].w>mid) sum[u]+=sum[v];
        else                 sum[u]+=Min(sum[v],tree[u][i].w);
    }
}

int main()
{
    while(scanf("%d %d",&n,&m),n+m) {
        init();
        int u,v,w;
        int l=inf,r=0;
        for(int i=1;i<n;i++) {
            scanf("%d %d %d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
            l=Min(w,l);
            r=Max(w,r);
        }
        int ans=inf;
        while(l<=r) {
            mid=(l+r)>>1;
            dfs(1,-1);
            if(sum[1]<=m) {
                ans=Min(mid,ans);
                r=mid-1;
            }
            else l=mid+1;
        }
        printf("%d\n",ans==inf?-1:ans);
    }
    return 0;
}