选课（洛谷—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=0表示没有直接先修课（1<=ki<=N, 1<=si<=20）。

输出格式

只有一行，选M门课程的最大得分。

输入输出样例

输入 

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

输出 

13

思路：这道题题意就是让你选课，有的课需要先学其他的课才能学。这就形成了一种依赖，要学一种课程，必须要学一个必修课，而且题意还是求最大学分，所以就是树形背包。

树形背包：

设dp[i][j]表示选择以i为根的子树中j个节点。
u代表当前根节点，tot代表其选择的节点的总额。
*/
void dfs(int u,int tot)
{
    for(int i=head[x];i;i=e[i].next)
    {
        int v=e[i].to;
        for(int k=0;k<tot;k++)//这里k从o开始到tot-1，因为v的子树可以选择的节点是u的子树的节点数减一
            dp[v][k]=dp[u][k]+val[u];
        dfs(v,tot-1)
        for(int k=1;k<=tot;k++)
            dp[u][k]=max(dp[u][k],dp[v][k-1]);//这里是把子树的值赋给了根节点，因为u选择k个点v只能选择k-1个点。
    }
}

 

AC代码：

#include <stdio.h>
#include <string>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <iostream>
const int maxx=10010;
using namespace std;
int dp[maxx][maxx],vis[maxx],head[maxx],val[maxx];
int n,m,cnt=0;
struct node
{
    int to,next;
} edge[maxx];
void add(int x,int y)
{
    edge[++cnt].next=head[x];
    head[x]=cnt;
    edge[cnt].to=y;
}
void dfs(int u,int tot)
{
    if(tot<=0) return ;
    for(int i=head[u]; i; i=edge[i].next)
    {
        int v=edge[i].to;
        for(int j=0; j<tot; j++)
            dp[v][j]=dp[u][j]+val[v];
        dfs(v,tot-1);
        for(int k=1; k<=tot; k++)
            dp[u][k]=max(dp[u][k],dp[v][k-1]);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int a;
    for(int i=1; i<=n; i++)
    {
        scanf("%d%d",&a,&val[i]);
        if(a)
            add(a,i);
        if(!a)
            add(0,i);
    }
    dfs(0,m);
    printf("%d\n",dp[0][m]);
    return 0;
}