[bzoj十连测第二场 A]深邃

题目大意

一棵树每个节点有颜色。
请将树分成若干连通块，使每个连通块均包含黑点。最小化最大的连通块的大小。

二分

看到双最直接二分。
二分了一个k，接下来怎么判定？
设f[i]表示以i为根的情况：
1、f[i]为正数，表示以i为根的子树内除了i所在连通块均合法，i所在连通块包含黑点时i所处连通块大小的最小值。
2、f[i]为负数，表示以i为根的子树内除了i所在连通块均合法，i所在连通块不包含黑点时i所处连通块大小的相反数的最大值。
于是，我们树形dp，如果x是黑点，则所有f为负的与其一个连通块，然后f[x]为正。如果x是白点，那么我们找到其子树中f为正最小的，如果带上这个子树大小不超过k则带上f[x]为正，否则不能带上f[x]为负，具体见代码。
注意存在任意一个f[x]其绝对值超过k就非法。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=200000+10;
int f[maxn],h[maxn],go[maxn*2],next[maxn*2];
bool bz[maxn],czy;
int i,j,k,l,r,mid,t,n,m,tot,top,ans,x,y,z,u,v,w;
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    int t=h[x],k=1<<30,l=0;
    while (t){
        if (go[t]!=y){
            dfs(go[t],x);
            if (f[go[t]]>0) k=min(k,f[go[t]]);else l-=f[go[t]];
        }
        t=next[t];
    }
    if (bz[x]) f[x]=l+1;
    else if (k+l+1<=mid) f[x]=k+l+1;
    else f[x]=-l-1;
    if (abs(f[x])>mid) czy=0;
}
bool check(){
    czy=1;
    dfs(1,0);
    return (f[1]>=0&&f[1]<=mid)&&czy;
}
int main(){
    freopen("deep.in","r",stdin);freopen("deep.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,n-1){
        scanf("%d%d",&j,&k);
        add(j,k);
        add(k,j);
    }
    fo(i,1,m){
        scanf("%d",&j);
        bz[j]=1;
    }
    l=1;r=n;
    while (l<r){
        mid=(l+r)/2;
        if (check()) r=mid;else l=mid+1;
    }
    printf("%d\n",l);
}