HDU 1520 Anniversary party (树形dp

树形dp入门

给出子树与结点的关系和权值，现在我们需要求出每个结点只能选取自身或者其父节点 问权值最大是多少

//
// Created by 22934 on 2018/7/31.
//
#include <bits/stdc++.h>

using namespace std;

#define CLR(a,b) memset(a,(b),sizeof(a))
#define ll long long
#define ls o<<1
#define rs o<<1|1
#define pb push_back
typedef unsigned long long ull;

const int MAXN = (int)1e5+10;

//按照关系进行建树， 每个结点有两种状态，参加和不参加
//dp[i][0] 代表以i为根的子树且i不参加能得到的最大活跃值， dp[i][1] 表示i参加能得到的最大活跃值

//dp[i][0] += max(dp[j][1], dp[j][0]);
//dp[i][1] += dp[j][0];

int dp[MAXN][2], a[MAXN];
bool vis[MAXN];
vector<int> E[MAXN];
int n;

void dfs(int u) {
    for(int i = 0; i < (int)E[u].size(); ++i) {
        int v = E[u][i];
        dfs(v);
        dp[u][0]  += max(dp[v][1], dp[E[u][i]][0]);
        dp[u][1] += dp[v][0];
    }
}
int main() {
    while(~scanf("%d",&n) &&  n) {
        CLR(vis, false); CLR(dp, 0);
        for(int i = 1; i <= n; ++i) {
            scanf("%d",&dp[i][1]);
        }
        int u, v;
        while(~scanf("%d%d",&u,&v) && (u||v)) {
            E[v].pb(u); vis[u] = true;
        }
        int rt;
        for(int i = 1; i <= n; ++i) {
            if(!vis[i]) rt = i;
        }
        dfs(rt);
        printf("%d\n",max(dp[rt][0], dp[rt][1]));
        for(int i = 1; i <= n; ++i)  E[i].clear();
    }
    return 0;
}