1486. 选课

题目描述

大学里实行学分。每门课程都有一定的学分，学生只要选修了这门课，并通过考核就能获得相应的学分。学生最后的学分是他各门课学分的总和。每个学生都要选择规定数量的课程。其中有些课程可以直接选修，有些课程需要一定的基础知识，必须在选了其他的一些课程的基础上才能选修。

例如，《剥皮术》就必须在选修了《屠龙术》后才能选修。

我们称《屠龙术》是《剥皮术》的先修课。

每门课的直接先修课最多之有一门。两门课也可能存在相同的先修课。

每门课都有一个课号，课号依次是1,2,3……。以下表为例说明。

课号 先修课号 学分

1 无 1

2 1 1

3 2 3

4 无 3

5 2 4

上表中1是2的先修课，即如果要选修2，则1必须已被选过。

同样，要选修3,那么1和2都一定被选修过。

每个学生可选的课程总数是一定的，请找出一种方案，使得到的总学分最多。

输入

第一行包括两个正整数M、N（中间一个空格），其中M表示总课程数（1<=M<=1000），N表示每个学生最多可选的课程总数。（1<=N<=M）。

以下M行每行代表一门课，课号依次是1,2,…,M。每行两个数，第一个数为这门课的直接先修课的课号（若不存在则为0），第二个数为该门课的学分。学分是不超过10的正整数。

测试数据保证学生能够选满N门课。

输出

只有一个数，即最多可得的学分。

Code

#include<cstdio>
#include<cstring>
int max(int x,int y){return x > y ? x : y;}
struct node
{
    int data;
    node *next;
    node()
    {
        next = NULL;
    }
};
node head[301],*tail[301];
int f[301][301],s[301],n,m,p[301];
void dfs(int x)
{
    node *root;
    root = &head[x];
    node *r = root -> next;
    while(root -> next != 0)
    {
        root = root -> next;
        dfs(root -> data);
        s[x] += s[root -> data];
    }
    while(r != 0)
    {
        for(int k = s[x];k >= 1;k--)
            for(int j = 1;j <= k;j++)
                    f[x][k] = max(f[x][k],f[x][k - j] + f[r -> data][j]);
        r = r -> next;
    }
    for(int i = s[x];i >= 0;i--)
        f[x][i + 1] = f[x][i] + p[x];
    return ;
}
int main()
{
    memset(s,0,sizeof(s));
    memset(f,0,sizeof(f));
    int a,b;
    scanf("%d%d",&n,&m);
    for(int i = 0;i <= n;i++)
        tail[i] = &head[i];
    for(int i = 1;i <= n;i++)
    {
        scanf("%d%d",&a,&b);
        s[a]++;
        node *t = new node();
        t -> data = i;
        p[i] = b;
        tail[a] -> next = t;
        tail[a] = t;
    }
    dfs(0);
    printf("%d\n",f[0][m+1]);
    return 0;
}