没有上司的舞会 codevs 1380

上树DP，记忆化搜索。

本题老师讲的方法是直接树形DP，但是由于我对树并不够了解，什么dfs也不想尝试(虽然感觉自己可以搞)，于是搞了个结构体存点以及该点的信息，用f[i][j]作为记忆化数组。以后最好能用结构体就用结构体。有条理，集中，而且对于转java有点帮助(还不知道转不转)。

 代码：

#include <cstdio>
#include <algorithm>
#define N 6003
using namespace std;

struct Node{
    int dad,son[N],R,top;
}a[N];
int x,y,n,root;
int f[N][2];
int ffind(int p,int flag)
{
   if(f[p][flag]) return f[p][flag];
   int ans=0;
   if(flag)
   {
       ans=a[p].R;
       for(int i=1;i<=a[p].top;i++)
       {
           ans+=ffind(a[p].son[i],0);
       }
   }
   else
   {
       for(int i=1;i<=a[p].top;i++)
       {
           ans+=max(ffind(a[p].son[i],0),ffind(a[p].son[i],1));
       }
   }
   f[p][flag]=ans;
   return ans;
}

int main()
{
    scanf ("%d",&n);
    for(int i=1;i<=n;i++)
        scanf ("%d",&a[i].R);
    for(int i=1;i<n;i++)
    {
        scanf ("%d%d",&x,&y);
        a[x].dad=y;
        a[y].son[++a[y].top]=x;
    }
    for(root=1;root<=n;root++)
    {
        if(a[root].dad==0) break;
    }

    int ans=max(ffind(root,0),ffind(root,1));
    /*printf("\n");
    for(int i=1;i<=n;i++)
    {
        printf("%d %d\n",ffind(i,0),ffind(i,1));
    }*/
    printf("%d",ans);
    return 0;
}