BZOJ 2525 POI2011 DYN-Dynamite

Problem

BZOJ权限题，放洛谷链接。

Solution

参考了PoPoQQQ大爷的题解OAO

考虑二分答案，则只需判断对于给定覆盖范围，判断选m个点能否覆盖所有的点即可。

对于以一个节点为根的子树，有三种状态：
0.这棵子树中存在一个选择的点，这个选择的点的贡献还能继续向上传递
1.这棵子树中存在一个未被覆盖的关键点，需要一些选择的点去覆盖他
2.这棵子树中既没有能继续向上传递的选择的点也不存在未覆盖的关键点
是不是少了一种状态？如果这棵子树中既存在能继续向上传递的选择的点又存在未被覆盖的关键节点呢？
这种状态可以被归进第二种状态中，因为我们需要一个子树外的节点被选择去覆盖这个未覆盖的关键点，那么如果子树内的选择节点还可以覆盖子树外的某个关键节点，那么子树外的选择节点一定也可以覆盖这个关键节点，子树内的选择节点的贡献失去了意义，因此可以被归为状态2
如果是状态0，记录这个点的贡献还能向上传递多少
如果是状态1，记录子树中离根节点最远的未被覆盖的关键点的距离

至于贪心的正确性呢，因为如果我们从下往上dp，那么我们知道当下面的点已经解决完了，则我们尽量应该选择一个深度尽量小的点，因为它不需要再对更下层的点覆盖做贡献。注意特判一下根节点需不需要放。

Code

#include <cstdio>
using namespace std;
const int maxn=300010;
struct data{int v,nxt;}edge[maxn<<1];
int n,m,p,l,r,mid,tot,ans,a[maxn],sta[maxn],f[maxn],head[maxn];
template <typename Tp> inline void read(Tp &x)
{
    x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
inline int max(int x,int y){return x>y?x:y;}
inline void insert(int u,int v)
{
    edge[++p]=(data){v,head[u]};head[u]=p;
    edge[++p]=(data){u,head[v]};head[v]=p;
}
void dp(int x,int pre,int lim)
{
    int near=-1,far=a[x]?0:-1;
    for(int i=head[x];i;i=edge[i].nxt)
      if(edge[i].v!=pre)
      {
        dp(edge[i].v,x,lim);
        if(sta[edge[i].v]==0) near=max(near,f[edge[i].v]-1);
        else if(sta[edge[i].v]==1) far=max(far,f[edge[i].v]+1);
      }
    if(near<far)
    {
        if(far==lim) tot++,f[x]=lim,sta[x]=0;
        else f[x]=far,sta[x]=1;
    }
    else if(~near) f[x]=near,sta[x]=0;
    else f[x]=0,sta[x]=2;
}
int check(int k)
{
    tot=0;
    dp(1,0,k);
    if(sta[1]==1) tot++;
    return tot<=m;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    read(n);read(m);
    for(int i=1;i<=n;i++) read(a[i]);
    for(int i=1,x,y;i<n;i++){read(x);read(y);insert(x,y);}
    l=0;r=n;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(check(mid)) ans=mid,r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",ans);
    return 0;
}