[bzoj1217][贪心]消防局的设立

Description

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

Input

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

Output

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

Sample Input

6
1
2
3
4
5

Sample Output

2

题解

开始想着树形dp，然后敲了一个弱鸡代码发现挂了。。再仔细一想，贪心666
每个消防局可以管理2个点，那么我们隔4个点放消防局就是最优的！
设f[x]为x到最近的消防局的距离，然后特殊处理一下叶子节点
哦，开始的点如果最近的消防局＞3，那么也只能放一个消防局了

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{
    int x,y,next;
}a[210000];int len,last[210000];
int ans;
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int f[110000],n;
void treedp(int x,int fa)
{
    int maxx=-999999999,minn=999999999;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            treedp(y,x);
            maxx=max(maxx,f[y]);minn=min(minn,f[y]);
        }
    }
    if(maxx+minn<=3)f[x]=minn+1;//两个都能到，那么优先选择最小的  
    else f[x]=maxx+1;
    if(maxx==-999999999)f[x]=3;//叶子
    if(f[x]==5){f[x]=0;ans++;}
    if(f[x]>=3 && fa==-1)ans++; 
}
int main()
{
    scanf("%d",&n);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++)
    {
        int x;
        scanf("%d",&x);
        ins(i+1,x);ins(x,i+1);
    }
    ans=0;
    treedp(1,-1);
    printf("%d\n",ans);
    return 0;
}