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=0;
        child=father=brother=0;
    }
}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,0,sizeof(de));
        for(i = 1 ; 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 = 0;
        for(i = 1; i <= n ; i++)
        {
            if(!de[i])
            {
                dfs(i);
                ans = max(max(tr[i].nmax,tr[i].ymax),ans);
            }
        }
        cout<<ans<<endl;
    }

    return 0;
}