P2014 [CTSC1997] 选课树形DP

最新推荐文章于 2024-06-05 20:37:14 发布

窦尊

最新推荐文章于 2024-06-05 20:37:14 发布

阅读量101

点赞数

分类专栏：树形dp 算法

本文链接：https://blog.csdn.net/m0_56511205/article/details/126815362

版权

课程选择最大学分深度优先搜索算法依赖关系

关键词由CSDN通过智能技术生成

算法同时被 2 个专栏收录

21 篇文章 0 订阅

订阅专栏

树形dp

4 篇文章 0 订阅

订阅专栏

题目描述

在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有 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

输出 #1

解决代码：

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int N = 700;
int f[N][N];
int e[N],ne[N],w[N],h[N],idx;
int n,m;

void add(int a,int b)
{
	e[idx] = b;
	ne[idx] = h[a];
	h[a]= idx++; 
}

void dfs(int u)			//处理以u为根的树，f[u][k]表示以u为根的树，修k门课的最大值 
{
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j = e[i];
		
		dfs(j);
		
		for(int k=m;k>=0;k--)
		{
			for(int s=1;s<=k;s++)
			{				//		  f[u][k-s]表示j的兄弟节点修k-s门课 
				f[u][k] = max(f[u][k],f[u][k-s]+f[j][s]);
			}
		}
	}
	
	for(int i=m;i>=1;i--) f[u][i] = f[u][i-1] + w[u];
	
}

int main()
{
	memset(h,-1,sizeof h);
	
	cin>>n>>m;
	
	for(int i=1;i<=n;i++)
	{
		int a;
		cin>>a>>w[i];
		add(a,i);
	}
	
	m++;
	dfs(0);
	
	cout<<f[0][m];
	
	return 0;
}