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;
}