试题描述:ios
大学实行学分制。每门课程都有必定的学分,学生只要选修了这门课并经过考核就能得到相应学分。学生最后的学分是他选修各门课的学分总和。
每一个学生都要选择规定数量的课程。其中有些课程能够直接选修,有些课程须要必定的基础知识,必须在选了其余的一些课程基础上才能选修。例如《数据结构》必须在选修了《高级语言程序设计》后才能选修。咱们称《高级语言程序设计》是《数据结构》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。为便于表述每门课都有一个课号,课号依次为 1,2,3…
上例中课号1是课号2的先修课,即若是要先修课号2,则课号1一定已被选过。一样,若是要选修课号3,那么课号1和课号2都必定被选修过。
学生不可能学完大学开设的全部课程,所以必须在入学时选定本身要学的课程。每一个学生可选课程的总数是给定的。如今请你找出一种选课方案,使得你能获得的学分最多,而且必须知足先修课优先的原则。假定课程间不存在时间上的冲突。git
首先能一眼看出来,这些课程是一个树形结构,对于每个节点i,都须要判断它的全部儿子是否取,又往下取多少。数据结构
这样咱们把问题转化成了一个分组背包:对于每个节点,每一颗子树就是一组,每一组又m(也就是选课数)个物品,第i物品的体积是i,价值是f[j][i](j为当前儿子)spa
而后转移方程就能够变为f[i][j][k]表示在以i为根节点的子树中,第j组,取k个物品的最优值。设计
易获得:f[i][j][k]=max(f[i][j][k],f[i][j-1][k-q](q为枚举物品)+f[i][y][q])(y为子节点)code
而后,咱们发现能够把j的那一位压去,可是注意循环要倒着写。blog
#include <iostream> #include <cstdio> #include <algorithm> #include <cstdlib> #include <cstring> #define MAXN 1010 #define in(a) a=read() #define REP(i,k,n) for(int i=k;i<=n;i++) using namespace std; inline int read(){ int x=0,f=1; char ch=getchar(); for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1; for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0'; return x*f; } int n,m; int total,head[MAXN],to[MAXN<<1],nxt[MAXN<<1],f[MAXN][MAXN],s[MAXN]; inline void adl(int a,int b){ total++; to[total]=b; nxt[total]=head[a]; head[a]=total; return ; } inline void dfs(int i){ for(int e=head[i];e;e=nxt[e]){ dfs(to[e]); for(int j=m;j>=0;j--) REP(k,0,j) f[i][j]=max(f[i][j],f[i][j-k]+f[to[e]][k]); } if(i) for(int j=m;j>0;j--) f[i][j]=f[i][j-1]+s[i]; return ; } int main(){ in(n),in(m); int a,b; REP(i,1,n) in(a),in(s[i]),adl(a,i); dfs(0); cout<<f[0][m]; return 0; }