没有上司的舞会 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][];
 int ffind(int p,int flag)
 {
    if(f[p][flag]) return f[p][flag];
    int ans=;
    if(flag)
    {
        ans=a[p].R;
        for(int i=;i<=a[p].top;i++)
        {
            ans+=ffind(a[p].son[i],);
        }
    }
    else
    {
        for(int i=;i<=a[p].top;i++)
        {
            ans+=max(ffind(a[p].son[i],),ffind(a[p].son[i],));
        }
    }
    f[p][flag]=ans;
    return ans;
 }

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

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