bzoj1217 [HNOI2003]消防局的设立

题目

树上的贪心，为了极大利用空间，当两个节点之间距离为5的话最好。

#include<bits/stdc++.h>
#define N 1005
using namespace std;
int n,f[N],x,ans;
vector <int> edge[N];
void add(int x,int y)
{
    edge[x].push_back(y);
} 
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(int x)
{
    char buf[15];
    int len=0;
    while(x)buf[++len]=x%10+'0',x/=10;
    for(int i=len;i>=1;i--)putchar(buf[i]);
}
void DFS(int x,int fa)
{
    int mx=-1e9,mn=1e9;
    for(int i=0;i<edge[x].size();i++)
    {
        int u=edge[x][i];
        if(u==fa)continue;
        DFS(u,x);
        mx=max(f[u],mx),mn=min(f[u],mn);
    }
    if(mx+mn<=3)f[x]=mn+1;
    else f[x]=mx+1;
    if(mx==-1e9)f[x]=3;
    if(f[x]==5)ans++,f[x]=0;
    if(f[x]>=3&&fa==0)ans++;
}
int main()
{
    //freopen("in.txt","r",stdin);
    n=read();
    for(int i=2;i<=n;i++) 
    {
        x=read();
        add(i,x),add(x,i);
    }
    DFS(1,0);
    write(ans);
    return 0;
}