洛谷-2279 [HNOI2003]消防局的设立

题目描述
2020年，人类在火星上建立了一个庞大的基地群，总共有n个基地。起初为了节约材料，人类只修建了n-1条道路来连接这些基地，并且每两个基地都能够通过道路到达，所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话，我们称基地A到基地B的距离为d。
由于火星上非常干燥，经常引发火灾，人类决定在火星上修建若干个消防局。消防局只能修建在基地里，每个消防局有能力扑灭与它距离不超过2的基地的火灾。
你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时，消防队有能力及时扑灭火灾。
输入输出格式
输入格式：
输入文件名为input.txt。
输入文件的第一行为n （n<=1000），表示火星上基地的数目。接下来的n-1行每行有一个正整数，其中文件第i行的正整数为a[i]，表示从编号为i的基地到编号为a[i]的基地之间有一条道路，为了更加简洁的描述树状结构的基地群，有a[i]<i。
输出格式：
输出文件名为output.txt
输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

输入输出样例
输入样例#1：
6
1
2
3
4
5

输出样例#1：
2

解释：树DP状态太难写了，看见题解可以贪心，就贪了一次，每次选尽量深的节点安排消防站，同时把覆盖到的点打个标记。

#include<bits/stdc++.h>
#define maxn 1010
#define maxm 2010
using namespace std;
int head[maxn],to[maxm],nxt[maxm],h[maxn],fa[maxn],tot=0;
bool vis[maxn];
struct cmp{
    bool operator () (int &a,int &b){
        return h[a]<h[b];
    } 
};
priority_queue<int,vector<int>,cmp> q;
void add(int x,int y){
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
void dfs(int temp,int father,int depth){
    fa[temp]=father;
    h[temp]=depth;
    for(int j=head[temp];j;j=nxt[j]){
        if(to[j]==father) continue;
        dfs(to[j],temp,depth+1);
    }
}
void dfs2(int temp,int depth){
    if(depth>2) return;
    vis[temp]=true;
    for(int j=head[temp];j;j=nxt[j]) dfs2(to[j],depth+1);
}
int main(){
    int n,cnt,x,y,ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++) scanf("%d",&x),add(i+1,x),add(x,i+1);
    dfs(1,0,1);
    for(int i=1;i<=n;i++) q.push(i);
    while(q.size()){
        while(q.size()&&vis[x=q.top()]) q.pop();
        if(!q.size()) break;
        if(fa[fa[x]]) dfs2(fa[fa[x]],0);
        else dfs2(1,0);
        ans++; 
    }
    printf("%d\n",ans);
    return 0;
}