BZOJ 2525: [Poi2011]Dynamite

Description

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

Solution

觉得怪怪的。。不知为何
有了权限号感觉还是很爽的
这题嘛，显然二分一发，再贪心
很吃惊PoPoQQQ大爷喜欢把dp叫做贪心
设二分到的答案是p
有个贪心的想法就是某子树最深的点往上p的距离应该要是一个点吧。
以这个点把所有离他范围p以内的点都删了(用dp的方法)
$O(n)$扫一发即可
总复杂度$O(nlogn)$

Code

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<time.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef vector<int> vec;
typedef priority_queue<int> pq;
#define pb push_back
#define ph push
#define fi first
#define se second
inline void Max(int &a,int b){if(a<b)a=b;}
inline void Min(int &a,int b){if(a>b||a==-1)a=b;}
template<class T>void rd(T &a){
    a=0;char c;
    while(c=getchar(),!isdigit(c));
    do a=a*10+(c^48);
        while(c=getchar(),isdigit(c));
}
template<class T>void nt(T x){
    if(!x)return;
    nt(x/10);
    putchar(48+x%10);
}
template<class T>void pt(T x){
    if(!x)putchar('0');
    else nt(x);
}
const int M=3e5+5;
int d[M],Mx,len;
vec G[M];
bool flag[M];
inline void predfs(int v,int fa){
    Max(Mx,d[v]=d[fa]+1);
    for(int i=0;i<G[v].size();++i){
        int to=G[v][i];
        if(to^fa)predfs(to,v);
    }
}
int md[M],w,rem[M],p,n,m;
inline void dfs(int v,int fa){
    int s=0;
    for(int i=0;i<G[v].size();++i){
        int to=G[v][i];
        if(to==fa)continue;
        dfs(to,v);
        Max(rem[v],rem[to]-1);
    }
    for(int i=0;i<G[v].size();++i){
        int to=G[v][i];
        if(to==fa)continue;
        if(md[to]-d[v]<=rem[v])md[to]=0;
        Max(md[v],md[to]);
    }
    if(rem[v]<0&&flag[v])Max(md[v],d[v]);
    if(md[v]-d[v]==p){
        ++w;
        rem[v]=p,md[v]=0;
        return;
    }
}
inline bool check(){
    memset(md,0,sizeof(md));
    memset(rem,-1,sizeof(rem));
    w=0;
    dfs(1,0);
    if(md[1])++w;
    return w<=m;
}
//#define LOCAL
int main(){
#ifdef LOCAL
    freopen("data.in","r",stdin);
    freopen("data.out","w",stdout);
#endif
    cin>>n>>m;
    for(int i=1;i<=n;++i)
        rd(flag[i]);
    for(int i=1,x,y;i<n;++i)
        rd(x),rd(y),G[x].pb(y),G[y].pb(x);
    predfs(1,1);
    int l=0,r=Mx,ans;
    for(int mid;l<=r;){
        p=mid=l+r>>1;
        if(check())r=mid-1,ans=mid;
        else l=mid+1;
    }
    p=1,check();
    cout<<ans<<endl;
    return 0;
}