rnqoj-30- [stupid]愚蠢的矿工-树形DP

把树转化为二叉树，然后再左右DP；

 

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
struct list
{
    int l;
    int r;
}node[2001];
int val[2001];
int vis[2001][101];
int m,n;
int dp(int x,int m)
{
    if(m==0)return 0;
    if(vis[x][m]!=-1)return vis[x][m];
    if(x==0)return 0;
    int ans=0;
    ans=dp(node[x].r,m);
    for(int i=0;i<m;i++)
    {
        ans=max(ans,dp(node[x].l,i)+val[x]+dp(node[x].r,m-i-1));
    }
    vis[x][m]=ans;
    return ans;
}
void init()
{
    int i;
    scanf("%d%d",&n,&m);
    memset(vis,-1,sizeof(vis));
    for(i=1;i<=n;i++)scanf("%d",&val[i]);
    for(i=0;i<=n;i++)
    {
        node[i].l=node[i].r=0;
    }
    for(i=1;i<=n;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        node[b].r=node[a].l;
        node[a].l=b;
    }
    cout<<dp(node[0].l,m)<<endl;
}
int main()
{
    init();
    return 0;
}