luogu 2279 [HNOI2003]消防局的设立 树形dp

就是细节多一些，思路都非常常规.

Code:

#include <bits/stdc++.h>
#define N 1005
#define inf 1061109567
#define ll long long
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int n,edges;
int hd[N],to[N<<1],nex[N<<1],f[N][7];
void add(int u,int v)
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;
}
void dfs(int x,int ff)
{
    f[x][2]=1;
    f[x][3]=f[x][4]=0;
    int sum=0,sum2=0,aa=0,bb=0;
    for(int i=hd[x];i;i=nex[i])
    {
        int v=to[i];
        if(v==ff) continue;
        dfs(v,x);
        f[x][2]+=min(min(min(f[v][0],f[v][1]), min(f[v][2], f[v][3])), f[v][4]);
        sum+=min(f[v][0], f[v][1]);
        sum2+=min(min(f[v][0], f[v][1]), min(f[v][2], f[v][3]));
        aa+=f[v][0];
        bb+=f[v][3];
    }
    if(abs(aa)<=n) f[x][3]=aa;
    else f[x][3]=inf;
    if(abs(bb)<=n) f[x][4]=bb;
    else f[x][4]=inf;
    for(int i=hd[x];i;i=nex[i])
    {
        int v=to[i];
        if(v==ff) continue;
        if(abs(sum)<=n) f[x][0]=min(f[x][0], sum+f[v][1]-min(f[v][1],f[v][0]));
        if(abs(sum2)<=n) f[x][1]=min(f[x][1], sum2+f[v][2]-min(min(f[v][0], f[v][1]), min(f[v][2], f[v][3])));
    }
    f[x][3]=min(min(min(f[x][0], f[x][1]), f[x][2]), f[x][3]);
    f[x][4]=min(f[x][3], f[x][4]);
}
int main()
{
    int i,j,k;
    // setIO("input");
    scanf("%d",&n);
    for(i=2;i<=n;++i)
    {
        int x;
        scanf("%d",&x),add(i,x),add(x,i);
    }
    memset(f,0x3f,sizeof(f));
    dfs(1,0);
    printf("%d\n",min(min(f[1][0], f[1][1]), f[1][2]));
    return 0;
}