洛谷 P2014 选课

原题：
题目描述

在大学里每个学生，为了达到一定的学分，必须从很多课程里选择一些课程来学习，在课程里有些课程必须在某些课程之前学习，如高等数学总是在其它课程之前学习。现在有 N 门功课，每门课有个学分，每门课有一门或没有直接先修课（若课程 a 是课程 b 的先修课即只有学完了课程 a，才能学习课程 b）。一个学生要从这些课程里选择 M 门课程学习，问他能获得的最大学分是多少？
输入格式

第一行有两个整数 N , M 用空格隔开。( 1≤N≤300 , 1≤M≤300 )

接下来的 N 行,第 I+1 行包含两个整数 ki​和 si​, ki​ 表示第I门课的直接先修课，si​ 表示第I门课的学分。若 ki=0k_i=0ki​=0 表示没有直接先修课（1≤ki≤N 1≤si​≤20）。
输出格式

只有一行，选 MMM 门课程的最大得分。
输入输出样例
输入 #1

7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2

输出 #1

13

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 301;

int dp[maxn][maxn];
int s[maxn];
vector<int> G[maxn];
int n,m;


int dfs(int cur)
{
    int node_cnt = 1;
    dp[cur][1] = s[cur];
    if (G[cur].empty())
    {
        return 1;
    }
    for (int i = 0; i < G[cur].size(); i++)
    {
        int cnt = dfs(G[cur][i]);
        for (int j = min(m + 1, node_cnt); j>0; j--)
        {
            for (int k = 1; k <= cnt; k++)
            {
                if (k + j > m + 1)
                {
                    break;
                }
                dp[cur][k + j] = max(dp[cur][k + j], dp[G[cur][i]][k] + dp[cur][j]);
            }
        }
        node_cnt += cnt;
    }
    return node_cnt;
}
int main()
{
    ios::sync_with_stdio(false);
    while(cin>>n>>m)
    {
        int k;
        for (int i=0;i<=n;i++)
        {
            G[i].clear();
        }
        memset(s, 0 ,sizeof(s));
        for (int i=1;i<=n;i++)
        {
            cin >> k >> s[i];
            G[k].push_back(i);
        }
        dfs(0);
        cout << dp[0][m+1]<<endl;
    }
    return 0;
}

思路：

树形背包的裸题，也是树形dp类型的入门题目。

树形dp，肯定是以一个子树作为一个子状态去考虑，这个问题也显然是这样的。
从背包模型上来看，物品的价值是课程的分数，物品的体积是1，代表一门课程。

从树形结构来考虑，有个天然的子结构，那就是一颗子树。
与正常的背包模型不同的是，如果考虑以u为根节点的子树，如果要求考虑其子节点v，如果要将v的状态合并到u中，那么也要将v的子孙节点一并合入到u中，类似于背包中的依赖物品的关系（参见背包九讲）。
此外，从状态考虑上来看，对于每个子树的根节点u都应该包含可以描述u的子树v状态的方程，因此设置状态方程为。
$d[u][i][j]=max(dp[u][i-1][j-k]+dp[v][s_v][k])$
表示以u为根节点，选择前i个子树，选了j门课程的最大学分，那么转移方程表示的是考虑第i个子树，选择第i个子树的中的k个课程的最大学分。

状态转移方程可以使用滚动数组实现，节省空间。即把第二维优化掉。因为当遍历第i个子树的时候一定是在1到i-1这些子树已经处理好的状态中继续计算的，所以第i个子树利用滚动数组的方式可以实现空间优化。

树形背包的入门题目，好好揣摩一下