bzoj 1217: [HNOI2003]消防局的设立

题意：

bzoj 4667的退化版

题解：

bzoj 4667
code：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
using namespace std;
const int inf=1000000000;
int n,d,w[500010];
bool mark[500010];
struct node{
    int y,next;
}a[1000010];int len=0,last[500010];
int f[500010][25],g[500010][25];//f[x][i]x走i步走不到的点覆盖。 
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void dfs(int x,int fa)
{
    if(mark[x]) f[x][0]=g[x][0]=w[x];
    for(int i=1;i<=d;i++) g[x][i]=w[x];
    g[x][d+1]=inf;
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa) continue;
        dfs(y,x);
        for(int j=d;j>=0;j--) g[x][j]=min(g[x][j]+f[y][j],g[y][j+1]+f[x][j+1]);
        for(int j=d;j>=0;j--) g[x][j]=min(g[x][j],g[x][j+1]);
        f[x][0]=g[x][0];
        for(int j=1;j<=d+1;j++) f[x][j]+=f[y][j-1];
        for(int j=1;j<=d+1;j++) f[x][j]=min(f[x][j],f[x][j-1]);
    }
}
int main()
{
    scanf("%d",&n);d=2;
    for(int i=1;i<=n;i++) w[i]=1;
    memset(mark,true,sizeof(mark));
    for(int i=2;i<=n;i++)
    {
        int x;scanf("%d",&x);
        ins(i,x);ins(x,i);
    }
    dfs(1,0);
    printf("%d",f[1][0]);
}