Information Disturbing

5 5
1 3 2
1 4 3
3 5 5
4 2 6
0 0

3

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define ll long long
#define N 1100
#define mem(a,t) memset(a,t,sizeof(a))
const int inf=1000005;
struct node
{
    int v,w,next;
}e[N*2];
int dp[N];
int head[N];
int vis[N];
int cnt,n;
void add(int u,int v,int w)
{
    e[cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}
void tree_dp(int u,int limit)
{
    int i,v,flag=0,tmp=0;       //flag标记是否是叶子节点
    for(i=head[u];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if(!vis[v])
        {
            vis[v]=1;
            flag=1;
            tree_dp(v,limit);
            if(e[i].w>limit)         //当前边权值大于限制，只能取dp[v]
                tmp+=dp[v];
            else                     //取最优解
                tmp+=min(dp[v],e[i].w);
        }
    }
    if(flag)          
        dp[u]=tmp;
    else            //叶子节点权值为inf
        dp[u]=inf;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int i,u,v,m,w;
    int l,r,mid,ans;
    while(scanf("%d%d",&n,&m),n||m)
    {
        mem(head,-1);
        cnt=0;
        l=r=1;
        for(i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
            r=max(r,w);
        }
        ans=-1;
        while(l<=r)
        {
            mid=(l+r)>>1;
            for(i=1;i<=n;i++)
                vis[i]=dp[i]=0;
            vis[1]=1;
            tree_dp(1,mid);        
            if(dp[1]<=m)
            {
                r=mid-1;
                ans=mid;
            }
            else
                l=mid+1;
        }
        printf("%d\n",ans);
    }
    return 0;
}

hdu 3586 Information Disturbing （树形dp）

原文：http://blog.csdn.net/u011721440/article/details/45391421