bzoj1217: [HNOI2003]消防局的设立

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

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