洛谷P2014 选课

首先分析题目，这是一道树形dp的题目，是树形背包类的问题，以为每门课的先修课只有一门，所以这一定可以

构成一个森林结构，于是我们可以设计一个虚拟的根节点作为森林的根。

状态转移方程如下

dp[v][k]=dp[u][k]+val

dp[u][k]=max(dp[u][k],dp[v][k−1])

#include<iostream>
#include<cstdio>
using namespace std;
struct edge
{
    int next;
    int to;
}g[10005];
int n,m,num;
int last[10005];
int f[10005];
int dp[1005][1005];
int val[10005];
int aa;
int a[10005];
void dfs(int u,int t)
{
    if(t==0) return;
    for(int i=last[u];i;i=g[i].next)
    {
        int v=g[i].to;
        for(int k=0;k<=t;k++)
        dp[v][k]=dp[u][k]+val[v];
        dfs(v,t-1);
        for(int k=1;k<=t;k++)
        dp[u][k]=max(dp[u][k],dp[v][k-1]);
    }
}
void add(int from,int to)
{
    g[++num].next=last[from];
    g[num].to=to;
    last[from]=num;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&aa,&val[i]);
        if(aa)
        add(aa,i);
        else
        add(0,i);
    }
    dfs(0,m);
    cout<<dp[0][m];
    return 0;
}