HDU - 1520 Anniversary party（树形dp基础）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1520

题目意思：每个点有权值，父节点和子节点不可以同时选，求最大权值和

思路：

dp[x][0]表示x这个点不选，dp[x][1]表示这个点选，to代表子节点，然后再dfs自底向上遍历一遍。

dp[x][0]+=max(dp[to][0],dp[to][1]);
dp[x][1]+=dp[to][0];

代码：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;

int n;
int a[6005];
vector<int>G[6005];
int dp[6005][2];
void dfs(int x){
    for (int i=0;i<G[x].size();i++){
        int to=G[x][i];
        dfs(to);
        dp[x][0]+=max(dp[to][0],dp[to][1]);
        dp[x][1]+=dp[to][0];
    }
}
int main() {
    while (~scanf ("%d",&n)){
        memset(dp,0,sizeof(dp));
        for (int i=0;i<6005;i++) G[i].clear();
        for (int i=1;i<=n;i++) {
            scanf ("%d",&a[i]);
            dp[i][1]=a[i];
        }
        int st=n*(n+1)/2;
        int u,v;
        while (~scanf ("%d%d",&u,&v)&&u&&v){
            G[v].push_back(u);
            st-=u;
        }
        dfs(st);
        printf ("%d\n",max(dp[st][0],dp[st][1]));
    }
    return 0;
}