洛谷 P2014 选课

P2014 选课

先将这棵树转化成二叉树

dp[i][j]表示以第i门科目为根节点的子树选j门科目最大的学分

记忆化搜索

k是枚举左儿子选的科目数

#include<bits/stdc++.h>
using namespace std;
#define maxn 1999
int n,m,fa[maxn],ls[maxn],rs[maxn],w[maxn],dp[maxn][maxn];

int dfs(int i,int j)
{
    if(i>n||j>m||i<1||j<1) return 0;
    if(dp[i][j]) return dp[i][j];
    for(int k=0;k<j;k++)
        dp[i][j]=max(dp[i][j],dfs(ls[i],k)+dfs(rs[i],j-k-1)+w[i]);
    dp[i][j]=max(dp[i][j],dfs(rs[i],j));
    return dp[i][j];
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d%d",&fa[i],&w[i]);
    for(int i=1;i<=n;i++)
    {
        if(!ls[fa[i]]) ls[fa[i]]=i;
        else{
            int dad=ls[fa[i]];
            while(rs[dad]) dad=rs[dad];
            rs[dad]=i;
        }
    }
    printf("%d\n",dfs(ls[0],m));
    return 0;
}