hdu3586，树形dp+二分

最新推荐文章于 2019-10-19 16:42:15 发布

sugar_lover

最新推荐文章于 2019-10-19 16:42:15 发布

阅读量381

点赞数

本文链接：https://blog.csdn.net/u014287583/article/details/22800941

版权

切断根与叶子结点之间的联系，所有切断的边不能超过一个值k，切断边的总和不能超过m，如果不存在这样的一个k，输出-1，否则输出k。

二分得到k，找到满足条件的最小值

#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;
}