洛谷-2014 选课

题目描述
在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有N门功课，每门课有个学分，每门课有一门或没有直接先修课（若课程a是课程b的先修课即只有学完了课程a，才能学习课程b）。一个学生要从这些课程里选择M门课程学习，问他能获得的最大学分是多少？
输入输出格式
输入格式：
第一行有两个整数N,M用空格隔开。(1<=N<=300,1<=M<=300)
接下来的N行,第I+1行包含两个整数ki和si, ki表示第I门课的直接先修课，si表示第I门课的学分。若ki=0表示没有直接先修课（1<=ki<=N, 1<=si<=20）。
输出格式：
只有一行，选M门课程的最大得分。

输入输出样例
输入样例#1：
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2

输出样例#1：
13

解释：树型DP+分组恰好背包，从这个选课关系，我们很容易得到一颗树，我们对这一颗树进行树型dp,dp[root][j]：以root为节点，选j门最大学分是多少，最后我们把每颗树看成一组，来一次分组恰好背包，直接输出答案就好了

#include<iostream>
#include<cstring>
#define N 333
using namespace std;
int head[N]={0};
int nex[N]={0};
int V[N]={0};
int tot=0;
int s[N]={0};
int du[N]={0};
int ret[N]={0};
int dp[N][N]={0};
int num[N]={0};
void add(int x,int y){
    tot++;
    nex[tot]=head[x];
    V[tot]=y;head[x]=tot;
}
int n=0,m=0;
void DP(int root){
    num[root]++;
    dp[root][1]=s[root];
    for(int i=head[root];i;i=nex[i]){
        int to=V[i];
        DP(to);
        for(int j=num[root];j>=1;j--){
            for(int k=num[to];k>=0;k--){
                dp[root][j+k]=max(dp[root][j+k],dp[root][j]+dp[to][k]);
            }
        }
        num[root]+=num[to];
    }
}
int main(){
    ios::sync_with_stdio(false);
    //freopen("/home/wxy/Desktop/test.in", "r", stdin);
    cin>>n>>m;
    for(int i=1,fa;i<=n;i++){
        cin>>fa>>s[i];
        if(fa){
            add(fa,i);
            du[i]++;
        }
    }
    for(int i=1;i<=n;i++){
        if(!du[i]){
            DP(i);
            for(int j=m;j>=0;j--){
                for(int k=0;k<=min(num[i],j);k++){
                    ret[j]=max(ret[j],ret[j-k]+dp[i][k]);
                }
            }
        }
    }
    cout<<ret[m]<<endl;
    return 0;
}