BZOJ2525: [Poi2011]Dynamite

我想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;
}