题目大意
一棵树每个节点有颜色。
请将树分成若干连通块,使每个连通块均包含黑点。最小化最大的连通块的大小。
二分
看到双最直接二分。
二分了一个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);
}