BZOJ 2645 陶陶吃苹果 [树形DP]

题目传送门

题解

一道树形DP SB题，vijos入门经典题，noip普及组T1前道题。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 4003
using namespace std;
int dp[N][503][2],a[N],ans;
int v[N][N],du[N],n,m,sum[N];
inline void read(int &res){
    static char ch;int flag=1;
    while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
void dfsf(int x){
    for(register int i=1;i<=du[x];i++) {
        int u=v[x][i];
        sum[u]=sum[x]+a[u];
        for(register int j=0;j<=m;j++)dp[u][j][0]=dp[x][j][0];
        dfsf(u);
        for(register int j=1;j<=m;j++)
            dp[x][j][0]=max(dp[x][j][0],dp[u][j-1][0]+a[u]);
    }
}
void dfss(int x){
    for(register int i=du[x];i>=1;i--) {
        int u=v[x][i];
        for(register int j=0;j<=m;j++)dp[u][j][1]=dp[x][j][1];
        dfss(u);
        for(register int j=1;j<=m;j++)
            dp[x][j][1]=max(dp[x][j][1],dp[u][j-1][1]+a[u]);
    }
}
int main(){
    read(n),read(m);
    for(register int x,i=1;i<=n;i++)
        read(x),read(a[i]),v[x][++du[x]]=i;
    sum[1]=a[1],dfsf(1),dfss(1);
    for(register int i=1;i<=n;i++)
        if(du[i]==0)for(register int j=0;j<=m;j++)
                ans=max(ans,dp[i][j][0]+dp[i][m-j][1]+sum[i]);    
    printf("%d\n",ans);
    return 0;
}