树规+二分 [Poi2011]Dynamite

Byteotian Cave的结构是一棵N个节点的树，其中某些点上面已经安置了炸药，现在需要点燃M个点上的引线引爆所有的炸药。
某个点上的引线被点燃后的1单位时间内，在树上和它相邻的点的引线会被点燃。如果一个有炸药的点的引信被点燃，那么这个点上的炸药会爆炸。
求引爆所有炸药的最短时间。

输入：
第一行是两个整数N,M。(1<=m<=n<=300000）
接下来一行有N个整数Di，第I个数为1表示该点有炸药。
接下来N-1行每行有两个数A,B，表示A和B之间有一条边。
输出：
最短时间。
样例解释：
点燃3,5上的引线。

很明显要二分答案。但接下来其实应该算是个贪心。
子节点要返回自己最深没有在规定时间内炸掉的儿子的深度以及最浅炸掉的儿子的深度。
然后有三种转移：
1.自己是炸药，被炸的节点距离>ans,就更新最深没炸为max(0,某个子节点的最大深度)；
2.子节点最浅+最深<=ans,互相之间可以“炸到”。
3.子节点最深==ans,那么当前节点就必须炸掉。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define N 300005
#define fir first
#define sec second
using namespace std;
typedef pair<int,int> D;
struct road{int v,next;}lu[N*2];
int n,m,e,sum,adj[N],a[N],f[N],g[N],sz[N];
void add(int u,int v){lu[++e]=(road){v,adj[u]};adj[u]=e;}
void gmin(int &x,int y){if(x>y)x=y;}
void gmax(int &x,int y){if(x<y)x=y;}
D check(int x,int fa,int mid)
{
    D s=D(N,-N);
    for(int i=adj[x];i;i=lu[i].next)
    {
        int to=lu[i].v;if(to==fa)continue;
        D h=check(to,x,mid);
        gmin(s.fir,h.fir+1);gmax(s.sec,h.sec+1);
    }   
    if(a[x]&&s.fir>mid)s.sec=max(0,s.sec);
    if(s.fir+s.sec<=mid)s.sec=-N;
    if(s.sec==mid)sum++,s.fir=0,s.sec=-N;
    return s;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    int l=0,r=n,mid;D hh;
    while(l<=r)
    {
        sum=0;mid=l+r>>1;
        hh=check(1,0,mid);
        if(hh.fir+hh.sec>mid)sum++;
        if(sum<=m)r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",r+1);
}