bzoj2097 [Usaco2010 Dec]Exercise 奶牛健美操（二分答案+贪心+树）

给你一棵树，你可以割掉S条边，问你割开之后得到的S+1棵树的直径的最大值最小是多少。
我们二分这个最大值mid，看如果要满足直径都小于等于mid需要至少割掉几条边。我们在树上贪心地求至少需要割几条边。记d[x]表示x子树中的点离fa[x]的最远距离。我们每次贪心的按儿子的d[x]从大往小割即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,S,h[N],num=0,d[N],mid,tot=0,a[N];
struct edge{
    int to,next;
}data[N<<1];
inline void dfs(int x,int fa){
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa) continue;dfs(y,x);
    }int cnt=0;d[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa) continue;a[++cnt]=d[y];
    }sort(a+1,a+cnt+1);
    for(int i=cnt;i>=1;--i){
        if(a[i]+a[i-1]>mid) ++tot;
        else{d[x]=a[i]+1;break;}
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();S=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }int l=0,r=n-1;
    while(l<=r){
        mid=l+r>>1;tot=0;dfs(1,0);
        if(tot<=S) r=mid-1;else l=mid+1;
    }printf("%d\n",r+1);
}