【洛谷P2279】[HNOI2003]消防局的设立

消防局的设立

题目链接

贪心：每次取出深度最大的节点，若没有被覆盖到，要想覆盖它，

最优的做法显然是将它的爷爷设为消防局

（因为该节点深度为最大，选兄弟、父亲所覆盖的节点，选了爷爷后都能够覆盖）

用优先队列维护深度即可

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define INF 100000
#define N 1010
int n,Head[N],num,ans,fa[N];
int ok[N],deep[N],vis[N];
struct cmp{
    bool operator ()(int a,int b){
        return deep[a]<deep[b];
    }
};
struct NODE{
    int to,next;
} e[N<<];
priority_queue< int, vector<int>, cmp > q;
inline int read(){
    int x=; char c=getchar();
    while(c<''||c>'') c=getchar();
    while(''<=c&&c<='') { x=(x<<)+(x<<)+c-''; c=getchar(); }
    return x;
}
inline void add(int x,int y){
    e[++num].to=y;
    e[num].next=Head[x];
    Head[x]=num;
}
void dfs(int t,int f){
    fa[t]=f;
    deep[t]=deep[f]+;
    for(int i=Head[t];i;i=e[i].next)
     if(e[i].to!=f) dfs(e[i].to,t);
}
void dfs2(int t,int dep){
    if(dep==) return;
    for(int i=Head[t];i;i=e[i].next){
        int v=e[i].to;
        vis[v]=;
        dfs2(v,dep+);
    }
}
int main()
{
    n=read(); int x;
    for(int i=;i<=n;i++){
        x=read();
        add(i,x); add(x,i);
    }
    dfs(,-);
    for(int i=;i<=n;i++) q.push(i);
    int u,gf;
    while(!q.empty()){
        u=q.top(); q.pop();
        if(vis[u]) continue;
        ans++;
        if(fa[u]!=-)
         if(fa[fa[u]]!=-)
          gf=fa[fa[u]];
            else gf=fa[u];
           else gf=u;
           vis[gf]=;
           dfs2(gf,);
    }
    printf("%d\n",ans);
    return ;
}