[树形DP][背包]选课

学校实行学分制。每门的必修课都有固定的学分,同时还必须获得相应的选修课程学分。学校开设了N(N <
300)门的选修课程,每个学生可选课程的数量M是给定的。学生选修了这M门课并考核通过就能获得相应的学分。
在选修课程中,有些课程可以直接选修,有些课程需要一定的基础知识,必须在选了其它的一些课程的基础上才能选修。例如《Frontpage》必须在选修了《Windows操作基础》之后才能选修。我们称《Windows操作基础》是《Frontpage》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。每门课都有一个课号,依次为1,2,3,…。

你的任务是为自己确定一个选课方案,使得你能得到的学分最多,并且必须满足先修课优先的原则。假定课程之间不存在时间上的冲突

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

以下N行每行代表一门课。课号依次为1,2,…,N。每行有两个数(用一个空格隔开),第一个数为这门课先修课的课号(若不存在先修课则该项为0),第二个数为这门课的学分。学分是不超过10的正整数。

输出 只有一行,选M门课程的最大得分。

样例输入
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
样例输出
13

分析:
背包类树形DP,这类DP的状态设计大多是第一维为节点,第二维为背包容量,结合了两类DP的状态设计方式。那么我们设 f [ i ] [ j ] f[i][j] f[i][j]为在i的子树中选j门课程的最高学分
初始化: f [ x ] [ 0 ] = 0 f[x][0]=0 f[x][0]=0
状态转移: f [ x ] [ t ] = m a x ( f [ y ] [ c ] ) + s c o r e [ x ] ( y ∈ s o n [ x ] a n d Σ c = t − 1 ) f[x][t]=max(f[y][c])+score[x] (y∈son[x] and Σc=t-1) f[x][t]=max(f[y][c])+score[x]yson[x]andΣc=t1
上面的方程还不涉及背包的内容,只是简单描述了树形结构上的转移,背包的转移和一般的背包类似,将一个子节点的f值视为价值,枚举容量,背包容量为t-1,就可以 O ( n 2 ) O(n^2) O(n2)做背包,然后树上的转移是 O ( n ) O(n) O(n)的,所以整体复杂度为 O ( n 3 ) O(n^3) O(n3)

注意节点0的容量是t

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=305,M=700;
int pt[N],vis[M],tot=0,head[M],nxt[M];
int f[N][N],score[N];
vector<int>son[N];
int n,m;
inline void add(int x,int y){vis[++tot]=y,nxt[tot]=head[x],head[x]=tot;}
void inson(int v){
	pt[v]=1;
	for(int i=head[v];i;i=nxt[i]){
		if(pt[vis[i]]) continue;
		son[v].push_back(vis[i]);
		inson(vis[i]);
	}
}
void dfs(int v){
	f[v][0]=0;
	for(int i=0;i<son[v].size();i++){
		int y=son[v][i];
		dfs(y);
		for(int t=m;t>=0;t--){
			for(int j=t;j>=0;j--){
				if(t-j>=0)
					f[v][t]=max(f[v][t],f[v][t-j]+f[y][j]);
			}
		}
	}
	if(v!=0)
		for(int t=m;t;t--)
			f[v][t]=f[v][t-1]+score[v];
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int x,y;
		scanf("%d%d",&x,&score[i]);
		add(i,x);
		add(x,i);
	}
	inson(0);
	dfs(0);
	int ans=0;
	for(int i=m;i>=0;i--) ans=max(ans,f[0][i]);
	cout<<ans<<endl;
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值