hdu3586，树形dp+二分

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#define INF 1000001
int dp[1005],first[1005],next[2005],mid;
struct edge
{
int u;
int v;
int w;
}en[2005];
int min(int a,int b)
{
return a>b?b:a;
}
void add(int a,int b,int c,int p)
{
en[p].u=a;
en[p].v=b;
en[p].w=c;
next[p]=first[a];
first[a]=p;
}
void getdp(int s,int fu)
{
int i,flag;
//printf("a%d\n",mid);
flag=0;
for(i=first[s];i!=-1;i=next[i])
{
if(en[i].v!=fu)
{
flag=1;
getdp(en[i].v,s);
if(en[i].w<=mid)
{
dp[s]+=min(dp[en[i].v],en[i].w);
}
else
{
dp[s]+=dp[en[i].v];
}
}
}
if(flag==0)
dp[s]=INF;
}
int main()
{
int n,m,i,j,k,l,le,ri,a,b,c,idx;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==0&&m==0)
break;
memset(first,-1,sizeof(first));
memset(next,-1,sizeof(next));
for(i=1,idx=0;i<n;i++)
{
scanf("%d%d%d",&a,&b,&c);
idx++;
add(a,b,c,idx);
idx++;
add(b,a,c,idx);
}
le=0;ri=1000;
while(ri>le)
{
mid=(ri+le)>>1;
memset(dp,0,sizeof(dp));
getdp(1,-1);
if(dp[1]<=m)
ri=mid;
else
le=mid+1;
/*for(i=1;i<=n;i++)
printf("%d ",dp[i]);
printf("\n");*/
}
memset(dp,0,sizeof(dp));
mid=le;
getdp(1,-1);
if(dp[1]<=m)
printf("%d\n",le);
else
printf("-1\n");
}
return 0;
}