【洛谷P2014】选课【树形DP】【背包】

版权声明：若希望转载，在评论里直接说明即可，谢谢！ https://blog.csdn.net/SSL_ZYC/article/details/83660259

题目大意：

题目链接：https://www.luogu.org/problemnew/show/P2014
有 $n$门功课，一些功课有先修课。每门功课都有学分。求选出 $m$门功课能获得的最大学分。

---

思路：

树形DP+背包。
很明显，这道题肯定是设 $f[u][j]$表示以 $u$为根的子树选出 $j$门课程学习能获得的最大学分。
那么对于 $u$的任意一棵子树 $v$，我们设它有 $s$个结点，那么我们就可以在这棵子树中选择任意 $k\in[0\sim s]$个结点来转移。
那么方程很明显就是
$f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k])$
为什么 $j-k$之后还要减 $1$呢？
因为要选择 $u$的子树，那么 $u$是肯定得选的，所以在转移的时候可以忽略掉这一个结点。
那么很明显，最终答案就是 $f[0][m]$。

---

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 400
using namespace std;

int n,m,f[N][N],head[N],tot;

struct edge
{
	int next,to,s;
}e[N]; 

void add(int from,int to,int s)
{
	e[++tot].to=to;
	e[tot].s=s;
	e[tot].next=head[from];
	head[from]=tot;
}

int dp(int x)
{
	int sum=1;  //以x为跟的节点个数
	for (int i=head[x];~i;i=e[i].next)
	{
		int sonsum=dp(e[i].to);  //子树节点个数
		sum=sum+sonsum;
		for (int j=sum-1;j>=0;j--)  //背包，要降序
		 for (int k=0;k<sonsum;k++)
		  if (j-k>=1)
		   f[x][j]=max(f[x][j],f[x][j-k-1]+f[e[i].to][k]);
	}
	return sum;
}

int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&m);
	int x,y;
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&y);
		f[i][0]=y;
		add(x,i,y);
	}
	dp(0);
	printf("%d\n",f[0][m]);
	return 0;
}