题意:
给出一颗n个结点的树,上面有若干个关键结点;
现在可以在这些结点上选最多m个点,求最小化关键点到选择点的最大距离;
题解:
首先这道题是一个最大最小化的问题,很容易想到二分;
二分一个数L表示答案的;
然后问题就转化成了一个判定性问题:判定能否用m个点覆盖整个树上的关键点;
判定过程是贪心的;
设dis[x]为x的子树中最近的选择的点的距离,g[x]为x的子树中最远的未覆盖的关键点的距离;
我们在深搜的过程中维护这两个东西;
当dis[x]+g[x]>mid时,说明x子树的点不能覆盖它子树的所有关键点了,这是我们就要考虑再选择一个点来覆盖树;
我们贪心的考虑的话,如果选择这个点的父亲能覆盖这个点,那么我们就让父亲来覆盖;
也就是只有满足g[x]>mid的时候,我们才新选择一个点;
这样的贪心是正确的,在二分判断的时候判断建的点是否<=m就可以了;
时间复杂度O(nlogn),似乎有些卡但是不太卡时;
代码:
#include<cctype>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 310000
#define LEN 1<<16
using namespace std;
typedef long long ll;
int next[N<<1],to[N<<1],head[N],ce;
int dis[N],g[N],cnt,mid;
bool a[N],is[N];
void add(int x,int y)
{
to[++ce]=y;
next[ce]=head[x];
head[x]=ce;
}
char getc()
{
static char *S,*T,buf[LEN];
if(S==T)
{
T=(S=buf)+fread(buf,1,LEN,stdin);
if(S==T)
return EOF;
}
return *S++;
}
int read()
{
static char ch;
static int D;
while(!isdigit(ch=getc()));
for(D=ch-'0';isdigit(ch=getc());)
D=D*10+ch-'0';
return D;
}
void dfs(int x,int pre)
{
g[x]=-0x3f3f3f3f;
dis[x]=0x3f3f3f3f;
is[x]=0;
for(int i=head[x];i;i=next[i])
{
if(to[i]!=pre)
{
dfs(to[i],x);
dis[x]=min(dis[x],dis[to[i]]+1);
g[x]=max(g[x],g[to[i]]+1);
is[x]=1;
}
}
if(a[x])
g[x]=max(g[x],0);
if(!is[x])
{
dis[x]=mid+1;
}
if(dis[x]+g[x]>mid&&g[x]+1>mid)
{
cnt++;
dis[x]=0;
g[x]=-0x3f3f3f3f;
}
if(dis[x]+g[x]<=mid)
{
g[x]=-0x3f3f3f3f;
}
}
int main()
{
int n,m,i,j,k,x,y,l,r;
n=read(),m=read();
for(i=1;i<=n;i++)
a[i]=read();
for(i=1;i<n;i++)
{
x=read(),y=read();
add(x,y),add(y,x);
}
l=0,r=n-1;
while(l<=r)
{
mid=l+r>>1;
cnt=0;
dfs(1,0);
if(cnt<m||(cnt==m&&dis[1]+g[1]<=mid))
r=mid-1;
else
l=mid+1;
}
printf("%d\n",l);
return 0;
}