luogu2279 消防局的设立 (贪心)

按点的深度从大到小排序，每次取出深度最大的那个点，如果它还没被覆盖，就在它爷爷上放一个消防局，这样一定是最优的

为了判定是否被覆盖，可以记录从某点的子树中到这个点的最近消防局的距离dis[]（如果没有就是inf呗）

这样的话，对于x，如果dis[x]<=2或者dis[fa[x]]<=1或者dis[fa[fa[x]]]<=0，他就已经被覆盖了，直接跳过

用这种方法，可以做覆盖范围更大的

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 const int maxn=,inf=1e9;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int N,md[maxn],fa[maxn],dep[maxn],id[maxn];

 inline bool cmp(int x,int y){return dep[x]>dep[y];}

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     N=rd();
     dep[]=id[]=;md[]=inf;
     for(i=;i<=N;i++){
         fa[i]=rd();id[i]=i;
         md[i]=inf;
         dep[i]=dep[fa[i]]+;
     }
     sort(id+,id+N+,cmp);
     int ans=;
     for(i=;i<=N;i++){
         int x=id[i];
         if(md[x]<=||(fa[x]&&(md[fa[x]]<=||(fa[fa[x]]&&md[fa[fa[x]]]<=)))) continue;
         int f=fa[x]?(fa[fa[x]]?fa[fa[x]]:fa[x]):x;
         md[f]=;
         ans++;
         if(fa[f]){
             md[fa[f]]=min(md[fa[f]],);
             if(fa[fa[f]])
                 md[fa[fa[f]]]=min(md[fa[fa[f]]],);
         }
     }
     printf("%d\n",ans);
     return ;
 }