luogu P2014 选课 树形背包

选课啊，好厉害的一道题，记得当年是琐大神给我讲的呢。

树形背包（废话）。

并不想再抄题干······

用DFS式的树形DP更新，能过的原因是这题数据范围是在是太小了。只有500；

如果说要注意的地方的话，就是dfs里的循环是逆序的，

原因同背包，从子树j中，不可能重复选。

（其实dp[i][k]是省去了一维的即从i的前j棵子树中选k门）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
int dp[500][500];
int a[500];
int fa[500];
struct edge{
	int to;int next;
};
edge e[100000+666];
int h[500];
int p;
inline void add(int a,int b)
{
	p++;
	e[p].next=h[a];
	e[p].to=b;
	h[a]=p;
	return;
}
int n,m;
inline int ra()
{
	int x=0;char ch=getchar();int flag=1;
	while(ch>'9'||ch<'0'){if(ch=='-')flag=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x*=10;x+=ch-'0';ch=getchar();}
	return x*flag;
}
void dfs(int now)
{
	dp[now][1]=a[now];
	for(int i=h[now];i;i=e[i].next)
	{
		dfs(e[i].to);
		for(int j=m+1;j>=1;j--)
		  for(int k=1;k<j;k++)
		  dp[now][j]=max(dp[now][j-k]+dp[e[i].to][k],dp[now][j]);
	}
}
int main()
{
	n=ra();m=ra();
	for(int i=1;i<=n;i++)
	{
	    fa[i]=ra();
	    if(fa[i]!=0)add(fa[i],i);
	    a[i]=ra();
	}
	for(int i=1;i<=n;i++)if(fa[i]==0)
	{
		fa[i]=n+1;
		add(n+1,i);
	}
	dfs(n+1);
	cout<<dp[n+1][m+1];
	return 0;
}