rnqoj-30- [stupid]愚蠢的矿工-树形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;  
}