bzoj1217: [HNOI2003]消防局的设立

传送门
显然在两个消防站之间距离为5时最优。
就dfs一遍, 再对根处理一下就可以了.。

#include<cstring>
#include<cmath>   
#include<cstdio>  
#include<iostream>  
#include<cstdlib>   
#include<algorithm>
#define N 1005
using namespace std;
struct edge{
    int to,next;
}e[2005];
int head[1005],f[1005],n,v,ans,tot;
inline void add(int x,int y){
    e[++tot].to=y;
    e[tot].next=head[x];
    head[x]=tot;
}
void dfs(int x,int fa){
    int mi=1000,ma=-mi;
    for (int i=head[x];i;i=e[i].next)
        if (e[i].to!=fa){
            dfs(e[i].to,x);
            mi=min(mi,f[e[i].to]);
            ma=max(ma,f[e[i].to]);
        } 
    if (mi+ma<=3) f[x]=mi+1;
        else f[x]=ma+1;
    if (mi==1000) f[x]=3;
    if (f[x]==5) ans++,f[x]=0;
    else if (f[x]>2&&fa<0) ans++;
}
int main(){
    scanf("%d",&n);
    for (int i=2;i<=n;i++){
        scanf("%d",&v);
        add(i,v);
        add(v,i);
    }
    dfs(1,-1);
    printf("%d",ans);
}