选课 树形dp+路径输出

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 2010
using namespace std;
int n,m,v[maxn],sum[maxn],son[maxn][maxn],s[maxn][3],f[maxn][maxn];
bool falg[maxn];
int Dfs(int k,int p)
{
if(k==0&&p!=m||p==0)return 0;
if(f[k][p])return f[k][p];
int s1=0,s2=0,lc=son[k][1],rc=son[k][2];
if(rc)s1=Dfs(rc,p);
for(int i=0;i<=p-1;i++)
s2=max(s2,Dfs(lc,i)+Dfs(rc,p-i-1)+v[k]);
return f[k][p]=max(s1,s2);
}
void Path(int k,int p)
{
int lc=son[k][1],rc=son[k][2];
if(rc&&f[rc][p]==f[k][p])
{
falg[k]=0;Path(rc,p);return;
}
for(int i=0;i<=p-1;i++)
{
if(f[k][p]==f[lc][i]+f[rc][p-i-1]+v[k])
{
falg[k]=1;Path(lc,i);
Path(rc,p-i-1);return;
}
}
}
int main()
{
freopen("course.in","r",stdin);
freopen("course.out","w",stdout);
scanf("%d%d",&n,&m);
int x,y;m++;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x,&y);v[i]=y;
if(son[x][1]==0)son[x][1]=i;
else
{
int now=son[x][1];
while(son[now][2])now=son[now][2];
son[now][2]=i;
}
}
int ans=Dfs(0,m);
printf("%d\n",ans);
Path(0,m);
for(int i=1;i<=n;i++)
if(falg[i])
printf("%d\n",i);
return 0;
}