洛谷 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：

612345

输出样例#1：

2

本来想练习树形dp的，结果，还是写了贪心。

可以证明，如果深度最大的点没有被覆盖，那么先覆盖它的爷爷（如果有）。

由于太晚了，导致了诸多错误，以后一定要早睡。。。

比如染色时，不是搜到一个染过色的就停，而是搜完距离小于等于2的所有点。

不说了，睡觉。

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1005;
int n,cnt,ans,dep[N],hd[N],fa[N];
bool b[N];
struct edge
{
    int to,nxt;
}v[2*N];
void addedge(int x,int y)
{
    ++cnt;
    v[cnt].to=y;
    v[cnt].nxt=hd[x];
    hd[x]=cnt;
}
void dfs(int u,int f)
{
    dep[u]=dep[f]+1;
    fa[u]=f;
    for(int i=hd[u];i;i=v[i].nxt)
        if(v[i].to!=f)
            dfs(v[i].to,u);
}
void col(int u,int f,int stp)
{
    if(stp==3)
        return ;
    b[u]=1;
    for(int i=hd[u];i;i=v[i].nxt)
        if(v[i].to!=f)
            col(v[i].to,u,stp+1);
}
int main()
{
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        addedge(i,x);
        addedge(x,i);
    }
    dfs(1,0);
    while(1)
    {
        //找到深度最大的没有被覆盖的点。 
        int k=-1,mx=0;
        for(int i=1;i<=n;i++)
            if(!b[i]&&dep[i]>mx)
                mx=dep[i],k=i;
        if(k<0)
            break;
        ans++;
        //找到该点的爷爷节点 
        if(!fa[k])
            col(k,0,0);
        else if(!fa[fa[k]])
            col(fa[k],0,0);
        else
            col(fa[fa[k]],0,0);
    }
    printf("%d\n",ans);
    return 0;
}