其他分享
首页 > 其他分享> > [CTSC1997] 选课

[CTSC1997] 选课

作者:互联网

一道典型的树型dp

首先每门课肯定只能选择一次,那么毫无疑问是一道树上的01背包,容量可以看作科目的门数

设计状态:\(f[i][j]代表了以i为根节点的子树中选择j门课程获得的最大得分\)

转移:对于节点\(x\),进行01背包的转移(记得倒序!记得倒序!记得倒序!)

小细节:对于每个节点的转移,若它不是0节点,要将这个节点的学分再加上

贴码

#include <bits/stdc++.h>
#define pb push_back
using namespace std;
const int N=310;
int n,m,s[N],f[N][N];//f[i][j]代表以i为根的子树中取j个节点取得的最大价值
vector<int> v[N];
void dp(int x)
{
	f[x][0]=0;
	for(int i=0;i<v[x].size();i++)
	{
		int to=v[x][i];
		dp(to);
		for(int t=m;t>=0;t--)//当前选课的总门数
			for(int j=t;j>=0;j--)//子节点选课门数
				if(t-j>=0)
					f[x][t]=max(f[x][t],f[x][t-j]+f[to][j]);
	}
	if(x!=0)
		for(int t=m;t>0;t--)
			f[x][t]=f[x][t-1]+s[x];//若非0节点,就要加上它本身的学分
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x>>s[i];
		v[x].pb(i);
	}
	memset(f,0xcf,sizeof(f));
	dp(0);
	cout<<f[0][m]<<endl;
	return 0;
}

标签:CTSC1997,选课,int,--,倒序,节点,dp
来源: https://www.cnblogs.com/sheepcsy/p/16420142.html