题意:N个人组成一棵树,编号1-N,其中编号为1的人为总司令,叶节点为前线士兵。点与点之间的边有一定的权值,代表砍掉这条边所要花费的代价。题目要求砍掉某些边使得所有的叶节点都无法与根节点联通,而且在满足砍掉的所有的边的和<=m的情况下的最小限制,这里的限制是指砍掉的边中的权值最大的数。
分析:树形DP+二分。二分限制limit,设sum[i]为砍掉以节点i为根节点的所有叶节点的费用总和,则有
if(tree[u][i].w>mid) sum[u]+=sum[v];
else sum[u]+=Min(sum[v],tree[u][i].w); v为节点u的子节点。
Code:
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <vector>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
//#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int maxn=1005;
const int inf=1000005;
struct edge {
int v,w;
};
vector<edge>tree[maxn];
int sum[maxn];
int n,m,mid;
void init() {
for(int i=1;i<=n;i++) tree[i].clear();
}
void add_edge(int u,int v,int w) {
edge e;
e.v=v;
e.w=w;
tree[u].push_back(e);
}
void dfs(int u,int fa) {
int sz=tree[u].size();
if(sz<=1 && tree[u][0].v==fa) {//根节点
sum[u]=inf;
return ;
}
sum[u]=0;
for(int i=0;i<sz;i++) {
int v=tree[u][i].v;
if(v==fa) continue;
dfs(v,u);
if(tree[u][i].w>mid) sum[u]+=sum[v];
else sum[u]+=Min(sum[v],tree[u][i].w);
}
}
int main()
{
while(scanf("%d %d",&n,&m),n+m) {
init();
int u,v,w;
int l=inf,r=0;
for(int i=1;i<n;i++) {
scanf("%d %d %d",&u,&v,&w);
add_edge(u,v,w);
add_edge(v,u,w);
l=Min(w,l);
r=Max(w,r);
}
int ans=inf;
while(l<=r) {
mid=(l+r)>>1;
dfs(1,-1);
if(sum[1]<=m) {
ans=Min(mid,ans);
r=mid-1;
}
else l=mid+1;
}
printf("%d\n",ans==inf?-1:ans);
}
return 0;
}