选课 洛谷P2014 树形dp

题目描述

在大学里每个学生,为了达到一定的学分,必须从很多课程里选择一些课程来学习,在课程里有些课程必须在某些课程之前学习,如高等数学总是在其它课程之前学习。现在有 NN 门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程 a 是课程 b 的先修课即只有学完了课程 a,才能学习课程 b)。一个学生要从这些课程里选择 MM 门课程学习,问他能获得的最大学分是多少?

输入格式

第一行有两个整数 NN , MM 用空格隔开。( 1 \leq N \leq 3001≤N≤300 , 1 \leq M \leq 3001≤M≤300 )

接下来的 NN 行,第 I+1I+1 行包含两个整数 k_iki​和 s_isi​, k_iki​ 表示第I门课的直接先修课,s_isi​ 表示第I门课的学分。若 k_i=0ki​=0 表示没有直接先修课(1 \leq {k_i} \leq N1≤ki​≤N , 1 \leq {s_i} \leq 201≤si​≤20)。

输出格式

只有一行,选 MM 门课程的最大得分。

输入输出样例

输入 #1复制

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

输出 #1复制

13
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 310;
int pre[MAXN],cnt,dp[MAXN][MAXN],grade[MAXN],m,n;
struct info{
	int to,nt;
}node[MAXN];
void add(int x,int y)
{
	cnt++;
	node[cnt].to = y;
	node[cnt].nt = pre[x];
	pre[x] = cnt;
}
void dfs(int now) //dp[i][j]表示以i为根节点选择j个结点的最大价值 dp[i][1]表示选择自己
{
	dp[now][1] = grade[now];
	for(int i = pre[now]; i ;i = node[i].nt){
		int t = node[i].to;
		dfs(t);
		for(int j = n;j > 0; j--){ //dp[i][j] 由dp[i][j-k]转移来的 所以j的遍历顺序要大到小
			for(int k = j-1;k > 0; k--){ 
				dp[now][j] = max(dp[now][j],dp[now][j-k]+dp[t][k]);
			}
		}
	}
}
int main()
{
	int v;
	cin >> m >> n;
	for(int i = 1;i <= m; i++){
		cin >> v >> grade[i];
		add(v,i);
	}
	n++;
	dfs(0);
	cout << dp[0][n];
	return 0;
}

上一篇:CSP20219——T2非零段划分


下一篇:牛客网视频总结4(矩阵、链表)