题目大意:一棵树,在一个点建消防局可以覆盖到其距离小于等于2的点,求覆盖所有点最少消防局数目
题解:跟蓝书上的一道题类似,考虑到叶节点需要覆盖,优先考虑深度最大的,覆盖离其距离为2的点,这样是最优的
我的收获:劲啊
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int M=10005;
const int B=2;
int n,t,ans,dep;
int head[M],fa[M];
bool vis[M];
vector <int> node[M];
struct edge{int to,nex;}e[M*2];
void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}
void dfs(int x,int f,int d){
node[d].push_back(x);dep=max(dep,d);
//为了避免排序,使用深度表
for(int i=head[x];i!=-1;i=e[i].nex){
int v=e[i].to;
if(v!=f) fa[v]=x,dfs(v,x,d+1);
}
}
void get(int x,int F,int l)
{
vis[x]=1;if(l==2) return;
for(int i=head[x];i!=-1;i=e[i].nex)
if(e[i].to!=F) get(e[i].to,x,l+1);
}
void work()
{
dfs(1,0,0);
for(int i=dep;i>=0;i--){
for(int j=0;j<node[i].size();j++){
int u=node[i][j];
if(vis[u]) continue;
for(int i=1;i<=B;i++) if(fa[u]) u=fa[u];
get(u,0,0);
ans++;
}
}
cout<<ans<<endl;
}
void init()
{
cin>>n;t=0;memset(head,-1,sizeof(head));
for(int i=2,x;i<=n;i++)
scanf("%d",&x),add(i,x),add(x,i);
}
int main()
{
init();
work();
return 0;
}