hud1520Anniversary party(树形DP）

链接

第一道树形DP

根据左儿子 右兄弟 将多叉树转化成二叉树 结构体里保存取这个节点和不取这个节点的最大值

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<stdlib.h>
 #include<algorithm>
 using namespace std;
 #define N 6010
 struct node
 {
     int child,father,brother;
     int nmax,ymax,v;
     void init()
     {
         nmax = ymax=;
         child=father=brother=;
     }
 }tr[N];
 int de[N];
 void dfs(int u)
 {
     int child = tr[u].child;
     while(child)
     {
         dfs(child);
         tr[u].ymax += tr[child].nmax;
         tr[u].nmax += max(tr[child].ymax,tr[child].nmax);
         child = tr[child].brother;
     }
 }
 int main()
 {
     int i,k,n,u,v;
     while(cin>>n)
     {
         memset(de,,sizeof(de));
         for(i =  ; i <= n ;i++)
         {
             cin>>k;
             tr[i].init();
             tr[i].ymax = k;
         }
         while(cin>>u>>v)
         {
             if(!u&&!v) break;
             tr[u].father = v;
             tr[u].brother = tr[v].child;
             tr[v].child = u;
             de[u]++;
         }
         int ans = ;
         for(i = ; i <= n ; i++)
         {
             if(!de[i])
             {
                 dfs(i);
                 ans = max(max(tr[i].nmax,tr[i].ymax),ans);
             }
         }
         cout<<ans<<endl;
     }

     return ;
 }