我又一次感受到了初始化的重要性:我又错在了初始化上了……(多组数据死全家!!!)
这题的思想还是挺好的,二分答案加上树形dp,分开来看都挺简单的,但合起来就需要一点脑洞了。
我们可以二分一个答案出来,作为题目要求的最大边权值,然后用树形dp来判断这个答案是否成立。
综上,这题就被完美的解决啦。
附上AC代码:
#include <cstdio>
#include <algorithm>
#include <cstring>
#define N 1010
using namespace std;
struct side{
int to,w,nt;
}s[N<<1];
int n,m,num,h[N],x,y,w,l,r,mid,ans;
inline void add(int x,int y,int w){
s[++num]=(side){y,w,h[x]},h[x]=num;
s[++num]=(side){x,w,h[y]},h[y]=num;
}
inline int so(int now,int fa){
int sum=0,p;bool b=0;
for (int i=h[now]; i; i=s[i].nt)
if (s[i].to!=fa){
b=1,p=so(s[i].to,now);
if (s[i].w>mid) sum+=p;
else sum+=min(s[i].w,p);
}
return b==1?sum:0xfffff;
}
int main(void){
while (scanf("%d%d",&n,&m),n+m){
memset(h,0,sizeof h),num=0,l=1,r=0;
for (int i=1; i<n; ++i)
scanf("%d%d%d",&x,&y,&w),add(x,y,w),r=max(r,w);
ans=2e9;
while (l<=r){
mid=(l+r)>>1;
if (so(1,-1)<=m) ans=mid,r=mid-1;
else l=mid+1;
}
printf("%d\n",ans==2e9?-1:ans);
}
return 0;
}