【洛谷 P2014&SSL 1606】[CTSC1997]选课【树形DP】

选课

Time Limit:1000MS Memory Limit:65536K

Description

题目传送

大学里实行学分。每门课程都有一定的学分，学生只要选修了这门课并考核通过就能获得相应的学分。学生最后的学分是他选修的各门课的学分的总和。
　　每个学生都要选择规定数量的课程。其中有些课程可以直接选修，有些课程需要一定的基础知识，必须在选了其它的一些课程的基础上才能选修。例如，《数据结构》必须在选修了《高级语言程序设计》之后才能选修。我们称《高级语言程序设计》是《数据结构》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。为便于表述每门课都有一个课号，课号依次为1，2，3，……。下面举例说明

上例中1是2的先修课，即如果要选修2，则1必定已被选过。同样，如果要选修3，那么1和2都一定已被选修过。
学生不可能学完大学所开设的所有课程，因此必须在入学时选定自己要学的课程。每个学生可选课程的总数是给定的。现在请你找出一种选课方案，使得你能得到学分最多，并且必须满足先修课优先的原则。假定课程之间不存在时间上的冲突。

Input

输入文件的第一行包括两个正整数M、N（中间用一个空格隔开）其中M表示待选课程总数（1≤M≤1000)，N表示学生可以选的课程总数（1≤N≤M)。
以下M行每行代表一门课，课号依次为1，2……M。每行有两个数（用一个空格隔开），第一个数为这门课的先修课的课号（若不存在先修课则该项为0），第二个数为这门课的学分。学分是不超过10的正整数。

Output

输出文件第一行只有一个数，即实际所选课程的学分总数。以下N行每行有一个数，表示学生所选课程的课号。

Sample Input

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

Sample Output

13

分析：

同样 由于课会影响别的课的选择 采用树形DP

这道题实际和二叉苹果树的思路比较像
不过二叉苹果树只是二叉树 而这道题可能为多叉树

我们就要将前面已经做过的作为左子树 当前点为右子树 转换成二叉树
$f[i][j]$表示以$i$为为根节点选$j$门课的最优值……
动态能量转移方程：$$f[i][j]=max(f[i][j],f[i][j-k-1]+f[q][k]+p[i][q])$$
其中 $q$表示$i$的子节点 $p$数组表示$i$的学分

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,f[305][305],son[305][305],e[305],v[305],p[305][305],ans,x,w;
void Tdp(int dep)
{
	e[dep]=1;
	for(int i=0;i<v[dep];i++)
	{
		int q=son[dep][i];  //子节点
		if(e[q]) continue;  //不往父节点搜
		Tdp(q);
		for(int j=m;j>=0;j--)
			for(int k=j-1;k>=0;k--)
				f[dep][j]=max(f[dep][j],f[q][k]+f[dep][j-k-1]+p[dep][q]);  //DP方程
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&w);
		p[x][i]=w;
		son[x][v[x]++]=i;
	}
	Tdp(0);
	printf("%d",f[0][m]);
	return 0;
}