描述
学校实行学分制。每门的必修课都有固定的学分,同时还必须获得相应的选修课程学分。学校开设了N(N<300)门的选修课程,每个学生可选课程的数量M是给定的。学生选修了这M门课并考核通过就能获得相应的学分。
在选修课程中,有些课程可以直接选修,有些课程需要一定的基础知识,必须在选了其它的一些课程的基础上才能选修。例如《Frontpage》必须在选修了《Windows操作基础》之后才能选修。我们称《Windows操作基础》是《Frontpage》的先修课。每门课的直接先修课最多只有一门。两门课也可能存在相同的先修课。每门课都有一个课号,依次为1,2,3,…。 例如:
表中1是2的先修课,2是3、4的先修课。如果要选3,那么1和2都一定已被选修过。 你的任务是为自己确定一个选课方案,使得你能得到的学分最多,并且必须满足先修课优先的原则。假定课程之间不存在时间上的冲突。
格式
输入格式
输入文件的第一行包括两个整数N、M(中间用一个空格隔开)其中1≤N≤300,1≤M≤N。
以下N行每行代表一门课。课号依次为1,2,…,N。每行有两个数(用一个空格隔开),第一个数为这门课先修课的课号(若不存在先修课则该项为0),第二个数为这门课的学分。学分是不超过10的正整数。
输出格式
输出文件每行只有一个数。第一行是实际所选课程的学分总数。
先可以确定此题是以树形结构存储,而且是森林,
所以需要转换成二叉树,但是注意转换成二叉树
后右子树原来是根的兄弟,所以可以不选根节点。
#include <stdio.h>
#include <string.h>
const int maxn = 405;
int son[maxn], dp[maxn][maxn];
struct node
{
int l, r, val;
} a[maxn];
inline int Max ( int a, int b )
{
return a > b ? a : b;
}
int dfs ( int i, int cnt )
{
if ( dp[i][cnt] >= 0 )
return dp[i][cnt];
if ( a[i].l < 0 && a[i].r < 0 )
{
if ( cnt == 1 )
dp[i][cnt] = a[i].val;
else
dp[i][cnt] = 0;
}
else if ( a[i].l < 0 && a[i].r >= 0 ) //注意右子树是兄弟
{
dp[i][cnt] = Max ( dp[i][cnt], dfs ( a[i].r, cnt-1 )+a[i].val );
dp[i][cnt] = Max ( dp[i][cnt], dfs ( a[i].r, cnt ) );
}
else if ( a[i].l >= 0 && a[i].r < 0 )
dp[i][cnt] = Max ( dp[i][cnt], dfs ( a[i].l, cnt-1 )+a[i].val );
else
{
dp[i][cnt] = Max ( dp[i][cnt], dfs ( a[i].r, cnt ) ); //先只选右子树
for ( int j = 1; j <= cnt; j ++ )
dp[i][cnt] = Max ( dp[i][cnt], dfs ( a[i].l, j-1 )+dfs ( a[i].r, cnt-j )+a[i].val );
}
return dp[i][cnt];
}
int main ( )
{
int n, m, last, sc;
scanf ( "%d%d", &n, &m );
memset ( dp, -1, sizeof ( dp ) ); //初始化为-1,结果0是个数为0时的答案
for ( int i = 0; i <= n; i ++ )
{
a[i].l = a[i].r = -1;
dp[i][0] = 0; //初始化
}
for ( int i = 1; i <= n; i ++ )
{
scanf ( "%d%d", &last, &sc );
a[i].val = sc;
if ( son[last] == 0 ) //多叉树转换成二叉树
a[last].l = i;
else
a[ son[last] ].r = i;
son[last] = i;
}
printf ( "%d", dfs ( 0, m+1 ) );
return 0;
}