HDU - 1520 树形DP入门题

写了两种DP，第一种是按照自己习惯来xjb敲的，第二种参考别人
熟悉一下树形DP的套路
dp[i][]是维护i及以下的关系最优值的，所以我觉得两次DP记忆搜索之间不清-1应该是正确的(也就做了一次加法)

#include<bits/stdc++.h>
#define rep(i,j,k) for(register int i=j;i<=k;i++)
#define rrep(i,j,k) for(register int i=j;i>=k;i--)
using namespace std;
typedef long long ll;
const int maxn  = 2e5+11;
int dp[maxn][2];
int w[maxn];
bool flag[maxn];
int to[maxn<<1],nxt[maxn<<1],head[maxn],tot;
void init(){memset(head,-1,sizeof head);tot=0;}
void add(int u,int v){
    to[tot]=v;nxt[tot]=head[u];
    head[u]=tot++;
}
int DP(int u,int now){
    if(~dp[u][now]) return dp[u][now];
    int ans=now?w[u]:0;
    for(int i = head[u]; ~i; i = nxt[i]){
        int v=to[i];
        if(now==0) ans+=max(max(DP(v,now),DP(v,now^1)),0);
        else ans+=max(0,DP(v,now^1));
    }
    return dp[u][now]=ans;
}
int DP(int u){
    dp[u][0]=0;
    dp[u][1]=w[u];
    for(int i = head[u]; ~i; i = nxt[i]){
        int v=to[i];
        DP(v);
        dp[u][0]+=max(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][0];
    }
    return max(dp[u][0],dp[u][1]);
}
int main(){
    int n,u,v,rt;
    while(scanf("%d",&n)!=EOF){
        init();memset(flag,0,sizeof flag);
        rep(i,1,n) scanf("%d",&w[i]);
        while(scanf("%d%d",&u,&v)!=EOF){
            if(u*v==0)break;
            add(v,u);flag[u]=1;
        }
        rep(i,1,n) if(!flag[i]){
            rt=i;
            break;
        }
        memset(dp,-1,sizeof dp);
        printf("%d\n",max(DP(rt,0),DP(rt,1)));
    }
    return 0;
}