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

时间:2021-08-01 09:16:40

消防局的设立

题目链接

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

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

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

用优先队列维护深度即可

#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 ;
}