[洛谷]P2014 选课 (#树形dp)

题目描述

在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有N门功课，每门课有个学分，每门课有一门或没有直接先修课（若课程a是课程b的先修课即只有学完了课程a，才能学习课程b）。一个学生要从这些课程里选择M门课程学习，问他能获得的最大学分是多少？

输入格式

第一行有两个整数N,M用空格隔开。(1<=N<=300,1<=M<=300)

接下来的N行,第I+1行包含两个整数ki和si, ki表示第I门课的直接先修课，si表示第I门课的学分。若ki=0表示没有直接先修课（1<=ki<=N, 1<=si<=20）。

输出格式

只有一行，选M门课程的最大得分。

输入输出样例

输入 #1复制

7  4
2  2
0  1
0  4
2  1
7  1
7  6
2  2

输出 #1复制

13

 

---

思路

树形dp。

每门课的直接先修课最多只有1门，将每门课的先修课与其相连，将会构成一颗或者多颗树。所以这会是一个森林。我们可以建立一个0号节点，将所有树的根节点都连向0号节点。0号节点的学分为0，并且允许选修的课数量要自加1。这样做的好处就是避免了最后还要链接每个根节点。然后我们会很方便对这棵以0为根节点的数进行树形dp。

对于节点i，它可以是0个或者1个或者多个节点的先修课，也就是父节点。令dp[i][j]为节点i选择j门课（包含自身）能获得的最大学分。选择节点i的子节点的前提是必须要选择节点i，所以令dp[i][1]=value[i]，其中value[i]为对于节点i的学分。

用邻接表存图后，遍历节点i的所有子节点。当遍历到节点son时（son是i的子节点，下面的代码中用v表示son），假设此时已经得到了所有dp[son][j]的值。由于此时dp[i][j]还没有son节点及其子树的学分，不妨通过son节点来更新i节点。则：

dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[son][k])

其中k为枚举选多少个节点，此时1<=j<=m+1，0<=k<=j-1。为什么k取不到j呢？因为j-k=0，就说明节点i无法取到！i是son的先修课，怎么可能取不到？所以k<j。

k是枚举多少个节点，要注意的是dp[i][j]是从dp[i][j-k]得到的，此时j要逆序枚举，因为如果顺推，在dp[i][j]上，显然已经转移过dp[i][j-k]，所以要保证更新dp[i][j]时dp[i][j-k]没被更新。

#include <stdio.h>
#include <iostream>
using namespace std;
int n,m,s,head[1001],cnt,dp[1001][1001],value[1001];
struct node
{
	int nxt,to;
}e[2001];
inline void add(int u,int v)
{
	e[++cnt].to=v;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs(int i)
{
	dp[i][1]=value[i];
	register int j,k,u;
	for(u=head[i];u;u=e[u].nxt)//遍历u的所有子节点 
	{
		int v(e[u].to);//子节点 
		dfs(v);
		for(j=m;j>=1;j--)
		{
			for(k=0;k<=j-1;k++)
			{
				dp[i][j]=max(dp[i][j],dp[i][j-k]+dp[v][k]);
			}
		}
	}
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j;
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		int u;
		cin>>u>>value[i];
		add(u,i);
	}
	m++;//由于0号节点的参与，m+1 
	dfs(0);
	cout<<dp[0][m]<<endl;
	return 0;
}