选课（树上背包）

题目描述

链接

学校实行学分制。

每门的必修课都有固定的学分，同时还必须获得相应的选修课程学分。

学校开设了 $N$ 门的选修课程，每个学生可选课程的数量 $M$ 是给定的。
学生选修了这 $M$ 门课并考核通过就能获得相应的学分。

在选修课程中，有些课程可以直接选修，有些课程需要一定的基础知识，必须在选了其他的一些课程的基础上才能选修。

例如《Windows程序设计》必须在选修了《Windows操作基础》之后才能选修。

我们称《Windows操作基础》是《Windows程序设计》的先修课。

每门课的直接先修课最多只有一门。

两门课可能存在相同的先修课。

你的任务是为自己确定一个选课方案，使得你能得到的学分最多，并且必须满足先修条件。

假定课程之间不存在时间上的冲突。

输入描述:

输入文件的第一行包括两个整数 $N、M$（中间用一个空格隔开）其中 $1\le N\le 300,1\le M\le N,1\le N\le 300,1\le M\le N$.

接下来 $N$ 行每行代表一门课，课号依次为 $1，2，\dots ，N$.

每行有两个数（用一个空格隔开），第一个数为这门课先修课的课号（若不存在先修课则该项为 $0$），第二个数为这门课的学分。

学分是不超过 $10$ 的正整数。

输出描述:

输出一个整数，表示学分总数。

输入

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

输出

13

思路

首先虚拟一个根节点，将森林转换为树。

设 $f[x][i][j]$ 为 $x$ 这棵子树上，根节点及前 $i$ 个分枝中学 $j$ 门课的最大学分。

那么对于第i个分枝（第i个子节点），有两种情况：

1. 这个分枝上一门课都不学.
2. 这个分枝上选修k门（ $0<k<j$ ）.

则：

$$f[x][i][j]=ma{x}_{0<k<j}\{\begin{array}{l}f[x][i-1][j]\\ f[x][i-1][j-k]+f[y][sizey][k]\end{array}$$

这里 $y$ 就是 $x$ 的第 $i$ 个分枝的根，$sizey$ 为 $y$ 的分枝的数量。

循环时取值的一些限制：

• 为什么要 $k<j$ 呢？
  与普通的背包问题不同，这个问题要求选某个课程的时候，必须先学它的前驱课。若是 $k==j$ 了，就意味着，在第 $i$ 个分枝上学k门课，在根节点及前 $i-1$ 课子树上一门也不学。这当然不行，至少要学根节点那一门课，否则就没有前驱课程了。

• 注意的一个地方：$1<j\le m$，$m$ 为最多可以选的课程数量。
  首先特判选则0门课，1门课的情况。之后的循环中，不可以再更新0门课、1门课的最值。
  因为选 $0$ 门课的学分为 $0$ ，没必要更新；选 $1$ 门课，只能选根节点，从题意上来说不可以更新最值。

省掉前 $i$ 个分枝那一维来节省空间，用倒序循环更新最值。

#include<bits/stdc++.h>
using namespace std;
constexpr int N=310;
vector<int> son[N];
int n,m,u,a[N],f[N][N];

void dfs(int x){
	f[x][0]=0,f[x][1]=a[x];
	for(int i=0;i<(int)son[x].size();i++){
		int y=son[x][i];
		dfs(y);
		for(int j=m;j>1;j--)
			for(int k=0;k<j;k++)
				f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
	}
}

int main(){
	cin>>n>>m;  m++;
	for(int i=1;i<=n;i++){
		cin>>u>>a[i];
		son[u].push_back(i);
	}
	memset(f,0xcf,sizeof(f));
	dfs(0);
	cout<<f[0][m]<<"\n";
}