其实就是把背包dp搬树上了,
注意每一个节点必须要先选自己才能选孩子节点
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const int inf=0x3f3f3f3f;
int n,m,val[305],f[305][305];//第i个节点选j个课程
bool b[305];
int tot,head[305],to[305],pre[305];
void addedge(int x,int y)
{
to[++tot]=y;pre[tot]=head[x];head[x]=tot;
}
void dfs(int u)
{
f[u][1]=val[u];//因为必须要保证先选根节点再选叶子节点!!
for (int i=head[u];i;i=pre[i])
{
int v=to[i];
dfs(v);
for (int j=m;j>=1;j--)//枚举的时候也不能出现不选根节点的情况 ,也就是说不能从0号开始枚举,其实这里j枚举的u节点现在选了几个点,l枚举v这个子节点选几个点,注意不能超了
{
for (int l=1;l<m;l++)
if (j+l<=m)
f[u][j+l]=max(f[u][j+l],f[u][j]+f[v][l]);
else break;
}
}
}
int main()
{
scanf("%d%d",&n,&m);m++;//因为0号节点,需要选
int u;
for (int i=1;i<=n;i++)
{
scanf("%d%d",&u,&val[i]);
addedge(u,i);
}
memset(f,-inf,sizeof(f));
dfs(0);
printf("%d",f[0][m]);
return 0;
}