先二分一个最短时间t,判断是否在最多选D个的情况下满足。
每个点设一个权值w,w<0表示其子树中(最远的未能在t时间引燃的炸药的离它的距离+1),w>0表示其子树中(炸药全部能引燃,而且还可以向上传递剩余的时间+1),w=0表示子树全部引燃但已经没有向上剩余。
设mxt表示其儿子中w的最大值,mit表示最小值,f[i]表示当前情况下子树中最少需要的引燃点数,转移如下:
if(mxt>=-mit) return max(mxt-1,0);
//最大的>=最小的,无需多设引燃点,但向上剩余-1。
if(mit==-len-1)
{
f[v]++;
return len;
}
//最小值达到了t,必须设一个引燃点,向上剩余变成t。
if(mxt<=0) return (mit-1);
//还未到t,最大深度差+1。
代码:
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=300010;
int n,m,f[maxn],dep[maxn],b[maxn];
struct edge
{
int t;
edge *next;
}*con[maxn];
void ins(int x,int y)
{
edge *p=new edge;
p->t=y;
p->next=con[x];
con[x]=p;
}
void dfs(int v,int fa)
{
dep[v]=dep[fa]+1;
for(edge *p=con[v];p;p=p->next)
if(p->t!=fa) dfs(p->t,v);
}
int cal(int v,int fa,int len)
{
int mxt=0,mit=-b[v];f[v]=0;
for(edge *p=con[v];p;p=p->next)
if(p->t!=fa)
{
int tmp=cal(p->t,v,len);
f[v]+=f[p->t];
mxt=max(mxt,tmp);
mit=min(mit,tmp);
}
if(mxt>=-mit) return max(mxt-1,0);
if(mit==-len-1)
{
f[v]++;
return len;
}
if(mxt<=0) return (mit-1);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
int x,y;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
ins(x,y);
ins(y,x);
}
dfs(1,0);
int l=0,r=n;
while(l<r)
{
int mid=(l+r)>>1,chk=(cal(1,0,mid)<0);
chk+=f[1];
if(chk<=m) r=mid;
else l=mid+1;
}
printf("%d",l);
return 0;
}