我想O(n)但是好像不太行?
于是二分一下答案
贪心的取标记点,每个标记点让他的层数尽量浅以对其他子树产生贡献
f[i]表示i的子树一定要多少个标记点,g[i]表示i的子树内最浅的标记点的层数,d[i]表示i的子树内未被覆盖的最深的点的层数
贪贪贪,dp一下
有一些细节qwq
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1e9
using namespace std;
inline void read(int &x)
{
char c; while(!((c=getchar())>='0'&&c<='9'));
x=c-'0';
while((c=getchar())>='0'&&c<='9') (x*=10)+=c-'0';
}
inline void up(int &x,const int &y){if(x<y)x=y;}
inline void down(int &x,const int &y){if(x>y)x=y;}
const int maxn = 310000;
int n,m;
int s[maxn];
struct edge{int y,nex;}a[maxn<<1]; int len,fir[maxn];
inline void ins(const int x,const int y){a[++len]=(edge){y,fir[x]};fir[x]=len;}
int dep[maxn],fa[maxn],mxd[maxn];
void search(const int x)
{
if(s[x]) mxd[x]=dep[x];
for(int k=fir[x],y=a[k].y;k;k=a[k].nex,y=a[k].y) if(y!=fa[x])
{
fa[y]=x; dep[y]=dep[x]+1;
search(y);
up(mxd[x],mxd[y]);
}
}
int mid;
int f[maxn],g[maxn],d[maxn];
void dfs(const int x,int br)
{
if(br>=mxd[x]-dep[x]) return;
d[x]=s[x]&&br<0?dep[x]:0;
if(mxd[x]<=dep[x]) return;
for(int k=fir[x],y=a[k].y;k;k=a[k].nex,y=a[k].y) if(y!=fa[x])
{
dfs(y,max(-1,br-1));
up(d[x],d[y]); f[x]+=f[y]; down(g[x],g[y]);
up(br,mid-(g[x]-dep[x]));
}
if(br>=d[x]-dep[x]) d[x]=0;
if(d[x]-dep[x]==mid) f[x]++,d[x]=0,g[x]=dep[x];
}
bool judge()
{
for(int i=1;i<=n;i++) f[i]=d[i]=0,g[i]=inf;
dfs(1,-1); if(d[1]>0) f[1]++;
return f[1]<=m;
}
int main()
{
read(n); read(m);
int num=0;
for(int i=1;i<=n;i++) read(s[i]),num+=s[i];
if(m>=num) return puts("0"),0;
for(int i=1;i<n;i++)
{
int x,y; read(x); read(y);
ins(x,y); ins(y,x);
}
dep[1]=1; search(1);
int l=1,r=n-1;
while(l<=r)
{
mid=l+r>>1;
if(judge()) r=mid-1;
else l=mid+1;
}
printf("%d\n",++r);
return 0;
}