[树形DP+贪心]BZOJ 1217—— [HNOI2003]消防局的设立 Plus？

题目描述

给出一棵树。

你可以设置一个特殊节点，距离这个节点小于等于2的节点被覆盖。

求覆盖所有节点所需的最小特殊节点数。

解题思路

这题是51 Nod 夹克老爷的愤怒的弱化版。

51Nod上的那题将范围推广至于n同阶的级别。

主要就是利用贪心进行树形DP。

#include<cstdio>
using namespace std;
const int maxn=100005,INF=2147483647;
int n,k,ans,f[maxn];
int tot,lnk[maxn],nxt[2*maxn],son[2*maxn];
inline int _read(){
    int num=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
int min(int x,int y){if (x<y) return x;return y;}
int max(int x,int y){if (x>y) return x;return y;}
void DFS(int x,int fa){
    int mi=INF,mx=-INF;
    for (int j=lnk[x];j;j=nxt[j])if (son[j]!=fa){
        DFS(son[j],x);
        mi=min(mi,f[son[j]]);
        mx=max(mx,f[son[j]]);
    }
    if (mi==INF) f[x]=-1;else
    if (mi<=-k){ans++;f[x]=k;}else
    if (mi+mx>0) f[x]=mx-1;
    else f[x]=mi-1;
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();k=2;
    for (int i=2;i<=n;i++){
        int x=_read();
        add(x,i);add(i,x);
    }
    DFS(1,0);
    printf("%d\n",ans+(f[1]<0));
    return 0;
}