P2279 [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

Solution：

　　本题和某次月赛的题目将军令简直一毛一样（～手动滑稽～）。

　　由于是一棵树，直接以$1$为根节点建树并处理出每个节点的深度，将节点按深度从大到小排序，每次贪心地去从最深的没有被标记的点往前走$4$步（因为控制半径为$2$，所以找出范围边界），经过的点打上标记，累加搜索次数就是答案。

代码：

#include<bits/stdc++.h>
#define il inline
#define ll long long
using namespace std;
const int N=10005;
int n,h[N],cnt,to[N],net[N],ans;
bool vis[N],ct[N];
struct point{
    int id,dis;
    bool operator <(point a){if(dis==a.dis)return id<a.id;return dis<a.dis;}
}a[N];
il void add(int x,int y){to[++cnt]=y,net[cnt]=h[x],h[x]=cnt;}
il void dfs(int u,int last){
    for(int i=h[u];i;i=net[i])
        if(last!=to[i]){
                a[to[i]].dis=a[u].dis+1;
                a[to[i]].id=to[i];
                dfs(to[i],u);
        }
}
il void dfs1(int u,int k){
    if(k<0)return;
    vis[u]=1,ct[u]=1;
    for(int i=h[u];i;i=net[i])
        if(!vis[to[i]])dfs1(to[i],k-1);
    vis[u]=0;
}
int main()
{
    ios::sync_with_stdio(0);
    cin>>n;
    int u;a[1].id=1,a[1].dis=0;
    for(int i=2;i<=n;i++)cin>>u,add(u,i),add(i,u);
    dfs(1,0);
    sort(a+1,a+n+1);
    for(int i=n;i>=1;i--)
        if(!ct[a[i].id]){
            ans++;
            dfs1(a[i].id,4);
        }
    cout<<ans;
    return 0;
}