hdu 3586 tree dp

来源:互联网 发布:乔治华盛顿大学 知乎 编辑:程序博客网 时间:2024/05/18 00:50

链接: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;}


0 0
原创粉丝点击