BZOJ2525 [Poi2011]Dynamite 题解（二分+贪心）

题目：BZOJ2525.
题目大意：给定一棵$n$个点的树和一些关键点，现在要求你选择$m$个点，最小化每个关键点到距离它最近的选择点的距离最大值，并输出最小值.
$1\le n,m\le 3\ast 10^5$.

首先凭借感觉二分答案，问题转化为判定用选择$m$个点是否能使得每个关键点到距离它最近的选择点距离不超过$mid$.

然后显然我们在不遗漏任何一个点的情况下，尽量让当前选择的点的深度尽量小，可以证明这个策略是最优的.

具体来说，对于某一棵子树的根$k$，我们记录子树中距离$k$最近的选择点到$k$的距离$di{s}_{0,k}$和子树中距离$k$最远的未被选择点覆盖到的关键点到$k$的距离$di{s}_{1,k}$.

此时，若有$di{s}_{0,k}+di{s}_{1,k}\le mid$，说明距离$k$最近的选择点可以覆盖距离$k$子树中所有未被覆盖的关键点，那么$di{s}_{1,k}=-\infty$.

同时，若$di{s}_{1,k}=mid$，说明这已经到了必须填的时候，这个时候就让答案$+1$，$di{s}_{0,k}=0,di{s}_{1,k}=-\infty$.

最后别忘了如果在根上$di{s}_{1,k}\ge 0$，需要再给答案加上$1$.

时间复杂度$O(n\log n)$.

代码如下：

#include<bits/stdc++.h>
using namespace std;

#define Abigail inline void
typedef long long LL;

const int N=300000,INF=(1<<30)-1; 

int n,m,a[N+9];
struct side{
  int y,next;
}e[N*2+9];
int lin[N+9],cs;

void Ins(int x,int y){e[++cs].y=y;e[cs].next=lin[x];lin[x]=cs;}
void Ins2(int x,int y){Ins(x,y);Ins(y,x);}

int dis[2][N+9];

int Dfs_dis(int k,int fa,int mid){
  int res=0;
  dis[0][k]=INF;dis[1][k]=-INF;
  for (int i=lin[k];i;i=e[i].next)
    if (e[i].y^fa){
      res+=Dfs_dis(e[i].y,k,mid);
      dis[0][k]=min(dis[0][k],dis[0][e[i].y]+1);
      dis[1][k]=max(dis[1][k],dis[1][e[i].y]+1);
    }
  if (a[k]&&dis[0][k]>mid) dis[1][k]=max(0,dis[1][k]);
  if (dis[0][k]+dis[1][k]<=mid) dis[1][k]=-INF;
  if (dis[1][k]==mid) dis[0][k]=0,dis[1][k]=-INF,++res;
  return res;
}

bool Check(int mid){return Dfs_dis(1,0,mid)+(dis[1][1]>=0)<=m;}

int Two_divide(){
  int l=0,r=n,mid=l+r>>1;
  for (;l<r;mid=l+r>>1)
    Check(mid)?r=mid:l=mid+1;
  return l;
}

Abigail into(){
  scanf("%d%d",&n,&m);
  for (int i=1;i<=n;++i)
    scanf("%d",&a[i]);
  for (int i=1;i<n;++i){
  	int x,y;
  	scanf("%d%d",&x,&y);
  	Ins2(x,y);
  }
}

Abigail outo(){
  printf("%d\n",Two_divide());
}

int main(){
  into();
  outo();
  return 0;
}