给一颗树,1为根节点,要切断1与所有叶子的联系,每条边有权值,给定权值总和上限,求在给定总和上限内切断单条边权值中最大值的最小值。
二分上限即可。存储边长的数组在空间有限制时可以在树节点中存储。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,idx;
int sum[1005],mx[1005],c[1005][1005];//可以换到下面
struct Node
{//int len;存储两节点间边长;
int v;
Node *next;
}*head[2010],tree[2010];
void init()
{
idx=1;
for(int i=0;i<=n;i++)
{
tree[i].v=i;
tree[i].next=NULL;
head[i]=NULL;
for(int j=0;j<=n;j++)
{
c[i][j]=0;
}
}
}
void add(int x,int y)
{
tree[idx].v=y;
tree[idx].next=head[x];
head[x]=&tree[idx++];
tree[idx].v=x;
tree[idx].next=head[y];
head[y]=&tree[idx++];
}
void Find(int rt,int par,int ul)
{
Node *p=head[rt];
while(p!=NULL)
{
if(p->v!=par)
{
if(mx[rt]==0)mx[rt]=c[rt][p->v];
else mx[rt]=max(mx[rt],c[rt][p->v]);
Find(p->v,rt,ul);
if(c[rt][p->v]<=ul)
{
if(mx[p->v]<=ul&&sum[p->v]!=0)
{
sum[rt]+=min(sum[p->v],c[rt][p->v]);
}
else sum[rt]+=c[rt][p->v];
}
else
{
if(mx[p->v]<=ul&&sum[p->v]!=0)
{
sum[rt]+=sum[p->v];
}
else sum[rt]=1000000;
}
if(sum[rt]>=1000000)sum[rt]=1000000;
}
p=p->next;
}
}
int main()
{
while(scanf("%d%d",&n,&m),n||m)
{
init();
int x,y,w;
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&w);
c[x][y]=w;
c[y][x]=w;
add(x,y);
}
int l=0,r=n,mid,ans=-1;
while(l<=r)
{
memset(sum,0,sizeof(sum));
memset(mx,0,sizeof(mx));
mid=(l+r)>>1;
Find(1,0,mid);
if(sum[1]<=m&&sum[1]!=0)
{
r=mid-1;
ans=mid;
}
else l=mid+1;
}
printf("%d\n",ans);
}
return 0;
}