P2014 选课（树形背包）

题目

https://www.luogu.org/problem/P2014

题意

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

思路

树形背包

代码

入门O(n*m*m)

#include <bits/stdc++.h>

using namespace std;

const int N = 100005;
int dp[1000][N];
int head[N],p;
struct node
{
    int v,nx;
}edge[N*2];
void add(int u,int v)
{
    edge[p].v = v;;
    edge[p].nx = head[u];
    head[u] = p++;
}
int n,m;
void ok(int u)
{
    for(int i = head[u];i != -1;i = edge[i].nx)
    {

        int v = edge[i].v;
        ok(v);
        for(int j = m;j >= 1;j--)
        {
            for(int k = 0;k < j;k++)
            {
                dp[u][j] = max(dp[u][j],dp[v][k] + dp[u][j-k]);
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    m++;
    memset(head,-1,sizeof(head));
    p = 0;
    for(int i = 1;i <= n;i++)
    {
        int u,x;
        scanf("%d%d",&u,&x);
        dp[i][1] = x;
        add(u,i);
    }
    ok(0);
    printf("%d\n",dp[0][m]);
    return 0;
}

 

 进阶O(n*m)

#include <bits/stdc++.h>

using namespace std;
const int maxn = 1e5+5;
int a[maxn],siz[maxn];
vector<int> dp[maxn];
vector<int> edge[maxn];
int n,m;
void dfs(int u)
{
    siz[u] = 1;
    dp[u][0] = 0;
    dp[u][1] = a[u];
    for(int i = 0;i < edge[u].size();i++)
    {
        int v = edge[u][i];
        dfs(v);
        int mi = min(siz[u],m);
        for(int i = mi;i >= 1;i--)
        {
            int mii = min(siz[v],m-i);
            for(int j = mii;j >= 1;j--)
            {
                dp[u][i+j] = max(dp[u][i+j],dp[u][i]+dp[v][j]);
            }
        }
        siz[u] += siz[v];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    m++;
    for(int i = 1;i <= n;i++)
    {
        int x;
        scanf("%d%d",&x,&a[i]);
        edge[x].push_back(i);
    }
    for(int i = 0;i <= n;i++) dp[i].resize(m+10);
    dfs(0);
    printf("%d\n",dp[0][m]);
    return 0;
}