树形DP专题

HDU1520 Anniversary party

给定棵树和每个节点的权值，相邻父子节点的不可重复选取，求其可以取到的最大权值

#include<bits/stdc++.h>
using namespace std;
const int MAX=6e3+5;
int n,a[MAX],d[MAX];
struct P
{
    int to,nxt;
}e[MAX];
int head[MAX],tot;
long long dp[MAX][2];
void init()
{
    memset(d,0,sizeof(d));
    memset(dp,0,sizeof(dp));
    memset(head,-1,sizeof(head));
    tot=0;
}
void adde(int u,int v)
{
    e[tot].to=v;
    e[tot].nxt=head[u];
    head[u]=tot++;
}
void dfs(int u)
{
    dp[u][0]=0;
    dp[u][1]=a[u];
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        dfs(v);
        dp[u][0]+=max(dp[v][1],dp[v][0]);
        dp[u][1]+=dp[v][0];
    }
}
int main()
{
    int x,y;
    while(~scanf("%d",&n))
    {
        init();
        for(int i=1;i<=n;++i) scanf("%d",&a[i]);
        while(~scanf("%d%d",&x,&y))
        {
            if(x==0&&y==0) break;
            adde(y,x);++d[x];
        }
        int rt=0;
        for(int i=1;i<=n;++i) if(!d[i]) {dfs(rt=i);break;}
        printf("%lld\n",max(dp[rt][0],dp[rt][1]));
    }
    return 0;
}