bzoj1217: [HNOI2003]消防局的设立

1217: [HNOI2003]消防局的设立

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 675   Solved: 365
[ Submit][ Status][ Discuss]

Description

2020年，人类在火星上建立了一个庞大的基地群，总共有n个基地。起初为了节约材料，人类只修建了n-1条道路来连接这些基地，并且每两个基地都能够通过道路到达，所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话，我们称基地A到基地B的距离为d。由于火星上非常干燥，经常引发火灾，人类决定在火星上修建若干个消防局。消防局只能修建在基地里，每个消防局有能力扑灭与它距离不超过2的基地的火灾。你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时，消防队有能力及时扑灭火灾。

Input

输入文件的第一行为n，表示火星上基地的数目。接下来的n-1行每行有一个正整数，其中文件第i行的正整数为a[i]，表示从编号为i的基地到编号为a[i]的基地之间有一条道路，为了更加简洁的描述树状结构的基地群，有a[i]

Output

输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

Sample Input

6
1
2
3
4
5

Sample Output

2

看到一个贪心题解有的思路：http://www.cnblogs.com/JSZX11556/p/5102075.html

另解可以用树状dp，不过弱弱的我不会QAQ。。。ps（按这个进度学吃枣药丸）

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <queue>
#include <vector>
#include <cmath>
#include <map>
#include <set>
using namespace std;
#define N 100000
#define INF 0x3f3f3f3f
struct edgenode{
    int to;
    int next;
}edge[N*4];
int head[N],vis[N],f[N];
int ans;
int cut;
void add(int u,int v)
{
    edge[cut].to=v;
    edge[cut].next=head[u];
    head[u]=cut++;
}
void dfs(int u,int fa)
{
    int maxi=-INF,mini=INF;
    for(int i=head[u];~i;i=edge[i].next) {
        int v=edge[i].to;
        if(v==fa)
            continue;
        dfs(v,u);
        mini=min(mini,f[v]);//距离自己最近的消防局
        maxi=max(maxi,f[v]);//距离自己最远的消防局
    }
    if(mini+maxi<=3)//能够覆盖到该点
        f[u]=mini+1;
    else            //不能覆盖到该点
        f[u]=maxi+1;
    if(maxi==-INF)
        f[u]=3;
    if(f[u]==5) {
        ans++;
        f[u]=0;
    }
    if(f[u]>=3&&fa==-1)
        ans++;
}
int main()
{
    int n;
    scanf("%d",&n);
    fill(head,head+N,-1);
    memset(f,0,sizeof(f));
    ans=0;

    for(int i=2;i<=n;i++){
        int x;
        scanf("%d",&x);
        add(x,i);
        add(i,x);
    }
    dfs(1,-1);
    printf("%d\n",ans);
}