1217: [HNOI2003]消防局的设立

题目链接

题目大意：一棵树，在一个点建消防局可以覆盖到其距离小于等于2的点，求覆盖所有点最少消防局数目

题解：跟蓝书上的一道题类似，考虑到叶节点需要覆盖，优先考虑深度最大的，覆盖离其距离为2的点，这样是最优的

我的收获：劲啊

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int M=10005;
const int B=2;

int n,t,ans,dep;
int head[M],fa[M];
bool vis[M];

vector <int> node[M];

struct edge{int to,nex;}e[M*2];

void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}

void dfs(int x,int f,int d){
    node[d].push_back(x);dep=max(dep,d);
//为了避免排序，使用深度表
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v!=f) fa[v]=x,dfs(v,x,d+1);
    }
}

void get(int x,int F,int l)
{
    vis[x]=1;if(l==2) return;
    for(int i=head[x];i!=-1;i=e[i].nex)
    if(e[i].to!=F) get(e[i].to,x,l+1);
}

void work()
{
    dfs(1,0,0);
    for(int i=dep;i>=0;i--){
        for(int j=0;j<node[i].size();j++){
            int u=node[i][j];
            if(vis[u]) continue;
            for(int i=1;i<=B;i++) if(fa[u]) u=fa[u]; 
            get(u,0,0);
            ans++;
        }
    }
    cout<<ans<<endl;
}

void init()
{
    cin>>n;t=0;memset(head,-1,sizeof(head));
    for(int i=2,x;i<=n;i++)
        scanf("%d",&x),add(i,x),add(x,i);
}

int main()
{
    init();
    work();
    return 0;
}