BZOJ2097: [Usaco2010 Dec]Exercise 奶牛健美操

n<=100000的树，砍S<n条边，求砍完后S+1棵树的最大直径的最小值。

树的直径要小小哒，那考虑一棵子树的情况吧！一棵子树的直径，就是子树根节点各儿子的最大深度+次大深度。就下面这样：

最大值最小肯定二分答案啦，那这棵子树如果有毛病呢，砍谁呢？肯定砍最大深度啦！所以就子树最大深度+次大深度有毛病就砍最大，没毛病把最大传上去。

写的优先队列，感觉排序会快一点？？

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
//#include<iostream>
using namespace std;

int n,S;
#define maxn 200011
struct Edge{int to,next;}edge[maxn];int first[maxn],le=2;
void in(int x,int y) {Edge &e=edge[le];e.to=y;e.next=first[x];first[x]=le++;}
void insert(int x,int y) {in(x,y);in(y,x);}
int lim,cnt;
int dfs(int x,int fa)
{
    priority_queue<int> q;
    for (int i=first[x];i;i=edge[i].next)
    {
        const Edge &e=edge[i];if (e.to==fa) continue;
        q.push(dfs(e.to,x));
    }
    while (!q.empty())
    {
        const int now=q.top();q.pop();
        if (!q.empty())
        {
            if (q.top()+now<=lim) {q.push(now);break;}
            cnt++;
        }
        else if (now<=lim) {q.push(now);break;}
        else cnt++;
    }
    return q.empty()?1:q.top()+1;
}
int x,y;
int main()
{
    scanf("%d%d",&n,&S);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        insert(x,y);
    }
    int L=1,R=n;
    while (L<R)
    {
        const int mid=(L+R)>>1;
        lim=mid,cnt=0;dfs(1,0);
        if (cnt<=S) R=mid;
        else L=mid+1;
    }
    printf("%d\n",L);
    return 0;
}