[BZOJ2525][Poi2011]Dynamite 二分答案+贪心+树型DP

先二分一个最短时间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;
}