【一本通提高树形动态规划】选课

最新推荐文章于 2024-06-12 10:00:23 发布

xueshenwudi

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量126

点赞数

文章标签：算法 c++ 动态规划

本文链接：https://blog.csdn.net/xueshenwudi/article/details/126532950

版权

题目链接

正解思路

设 $f[x][t]$ 表示以 $x$ 为根的子树中选 $t$ 门课程能够获得的最高学分，设 $x$ 的子节点集合为 $Son(x)$ ，子节点个数 $p=|son(x)|$ 修完 $x$ 这门课后，对于 $x$ 的每个子节点 $y\in son(x)$ ，我们可以在以 $y$ 为根的子树中选修若干门课（记为 $c_i$ ），在满足 $\sum c_i=t-1$ 的基础上获得尽量多的学分。

不难看出，动态转移方程为

$f[x][t]=0$

其中 $t=0$

$f[x][t]=max\{\sum_{i=1}^{p}f[y_i][c_i]\}+score[x]$

$\sum_{i=1}^{p} c_i = t-1$

其中 $t>0$

正解代码

#include<bits/stdc++.h>
using namespace std;
const int N=1512;
vector<int> son[N];
int f[N][N],s[N],n,m;
void dp(int x){
	int y;
	f[x][0]=0;
	for(int i=0;i<son[x].size();i++){
		y=son[x][i];
		dp(y);
		for(int t=m;t>-1;t--){
			for(int j=t;j>-1;j--){
				f[x][t]=max(f[x][t],f[x][t-j]+f[y][j]);
			}
		}
	}
	if(x!=0){
		for(int t=m;t>0;t--){
			f[x][t]=f[x][t-1]+s[x];
		}
	}
} 
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int x;
		cin>>x>>s[i];
		son[x].push_back(i);
	} 
	memset(f,0xc0,sizeof(f));
	dp(0);
	cout<<f[0][m];
	return 0;
}