【EOJ3654】管理孩子（贪心，二分）

题意：有一棵n个点的树，其中有k个是关键点，将树分割成若干部分，每部分至少包含一个关键点，求最大分割大小的最小值

思路：

最后特判一下f[1]的值

 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<map>
 #include<queue>
 #include<vector>
 #include<ctime>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define mem0(a) memset(a,0,sizeof(a))
 #define N  210000
 #define M  130
 #define MOD 1000000007
 #define eps 1e-8
 #define pi acos(-1)
 #define oo 1000000000

 int flag[N],f[N],a[N],n,mx;
 vector<int>c[N];

 void dfs(int u)
 {
     flag[u]=;
     if(a[u])
     {
         int t=;
         for(int i=;i<=(int)c[u].size()-;i++)
         {
             int v=c[u][i];
             if(!flag[v])
             {
                 dfs(v);
                 if(f[v]<) t-=f[v];
             }
         }
         f[u]=t+;
     }
      else
      {
          int t=,s=oo;
          for(int i=;i<=(int)c[u].size()-;i++)
          {
              int v=c[u][i];
              if(!flag[v])
              {
                  dfs(v);
                  if(f[v]<) t-=f[v];
                  else s=min(s,f[v]);
             }
         }
         if(mx-s>=t+) f[u]=s+t+;
          else f[u]=-t-;
      }
 }

 int isok(int m)
 {
     mem0(flag);
     mem0(f);
     mx=m;
     dfs();
     int flag=;
     if(f[]<) return ;
     for(int i=;i<=n;i++)
      if(abs(f[i])>m) return ;
     return ;
 }

 int main()
 {
     int k;
     scanf("%d%d",&n,&k);
     for(int i=;i<=n-;i++)
     {
         int x,y;
         scanf("%d%d",&x,&y);
         c[x].push_back(y);
         c[y].push_back(x);
     }
     for(int i=;i<=k;i++)
     {
         int x;
         scanf("%d",&x);
         a[x]=;
     }
     int l=,r=n,last=n;
     while(l<=r)
     {
         int mid=(l+r)>>;
         if(isok(mid)){last=mid; r=mid-;}
          else l=mid+;
     }
     printf("%d\n",last);
     return ;
 }