洛谷 P2014 [CTSC1997] 选课

题目描述
在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有 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_ik
i
​
和 s_is
i
​
, k_ik
i
​
表示第I门课的直接先修课，s_is
i
​
表示第I门课的学分。若 k_i=0k
i
​
=0 表示没有直接先修课（1 \leq {k_i} \leq N1≤k
i
​
≤N , 1 \leq {s_i} \leq 201≤s
i
​
≤20）。

输出格式
只有一行，选 MM 门课程的最大得分。

f[root][k]表示以root为根的树选k个节点的方案数

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
int n,m,a[1000],f[305][305];
vector<int> v[1000];

void dp(int root){

    for(int i=0;i<v[root].size();i++){

        int son=v[root][i];
        dp(son);
        for(int j=m+1;j>=1;j--){
            for(int k=0;k<j;k++){
                f[root][j]=max(f[root][j],f[root][j-k]+f[son][k]);
            }
        }
    }
}

int main(){
    ios::sync_with_stdio(0);
    cin>>n>>m;
    int x,y;
    for(int i=1;i<=n;i++){
        cin>>x>>y;
        v[x].push_back(i);
        f[i][1]=y;
    }

    dp(0);
    cout<<f[0][m+1];

    return 0;
}