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

题解

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 4003
using namespace std;
int dp
[503][2],a
,ans;
int v

,du
,n,m,sum
;
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;
}