洛谷P2014 选课

题目传送门

这道题可以比较容易地看出来是树形DP，将课程$x$的前置课程作为$x$的父亲节点，令$f[i][j]$表示在以节点$i$为根的子树中选择$j$门课程的最大学分。

但是有个问题：可以有好几门课没有前置课程，也就是说会有好几个根节点。对于这个问题，我们的解决办法是新建一个无前置课程、点权为0的0号点作为所有原来的根节点的父亲，这样就把几棵树连成了一棵，DP做完后我们只需要输出$f[0][m+1]$即可（0号点必须选）。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge
{
    int pre,to;
}e[510];
int n,m,f[510][510],h[510],top;
int dfs(int x)
{
    int i,j,k,v;
    for(i=h[x];i;i=e[i].pre)
    {
        v=e[i].to;dfs(v);
        for(j=m;j>1;j--)
            for(k=1;k<j;k++)
                f[x][j]=max(f[x][j],f[x][j-k]+f[v][k]);//枚举在以当前儿子为根的子树中选课的数量并更新
    }
    return 0;
}
int main()
{
    int i,j,x;
    scanf("%d%d",&n,&m);m++;//要先把m+1，不然f数组无法记录 
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&f[i][1]);
        e[++top].pre=h[x];h[x]=top;e[top].to=i;//连边 
    }
    dfs(0);cout<<f[0][m]<<endl;return 0;
}