BZOJ2097: [Usaco2010 Dec]Exercise 奶牛健美操 贪心+伪树dp+二分

//论全局变量的杀伤力....QAQ
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<vector>
#define MAXN 100005
using namespace std;
int f[MAXN];
struct Tree
{
   int to,next;
}c[MAXN<<];
vector<int>son[MAXN];
int head[MAXN],t;
inline void add(int x,int y)
{
   c[++t].to=y;
   c[t].next=head[x];
   head[x]=t;
   c[++t].to=x;
   c[t].next=head[y];
   head[y]=t;
}
int n,s,mid;
int comp(const int x,const int y)
{
   return f[x]>f[y];
}
void Dfs(int p,int fa)
{
   for(int i=head[p];i;i=c[i].next)
   if(c[i].to!=fa)
   {
     son[p].push_back(c[i].to);
     Dfs(c[i].to,p);
   }
   son[p].push_back();
}
void dfs(int p,int &x,int fa)
{
   for(int i=head[p];i;i=c[i].next)
   if(c[i].to!=fa)
     dfs(c[i].to,x,p);
   sort(son[p].begin(),son[p].end(),comp);
   int i=;
   for(;i<son[p].size()-;i++)
       if(f[son[p][i]]+f[son[p][i+]]>mid)
         x++;
       else
         break;
   f[p]=f[son[p][i]]+;
}
int main()
{
    scanf("%d%d",&n,&s);
    for(int i=;i<n;i++)
    {
       int x,y;
       scanf("%d%d",&x,&y);
       add(x,y);
    }
    Dfs(,);
    int l=,ans=n-,r=n-;
    while(l<=r)
    {
       mid=(l+r)>>;
       int x=;
       dfs(,x,);
       if(x<=s)
        ans=mid,r=mid-;
       else
        l=mid+;
    }
    printf("%d",ans);
    return ;
}