HDU 1520 Anniversary party （树形DP，入门）

 

 

题意：给一棵树，每个节点都有权值，要求选择部分节点出来，使得权值之和最大，但是每对（父亲，儿子）中最多只能挑一个。

 

思路：

　　比较入门的题，每个节点可以选也可以不选。若当前节点选的话，孩子必须全部不选；若当前节点不选，则孩子可以选也可以不选。

 

 

#include <bits/stdc++.h>
#define pii pair<int,int>
#define INF 0x3f3f3f3f
#define LL long long
using namespace std;
const int N=7010;
int w[N], dp[N][2], pre[N];
vector<int> tree[N];

void DFS(int t)
{
    for(int i=0; i<tree[t].size(); i++)    DFS(tree[t][i]);
    int sum[2]={0,0};
    for(int i=0; i<tree[t].size(); i++)
    {
        sum[0]+=max(dp[tree[t][i]][0], dp[tree[t][i]][1]); //点t尝试不取，则孩子可以取也可以不取，取大者即可。
        sum[1]+=max(dp[tree[t][i]][0], 0);    //点t尝试取，则孩子必须全部不取
    }
    dp[t][0]=sum[0];
    dp[t][1]=sum[1]+w[t];
}


int main()
{
    //freopen("input.txt", "r", stdin);
    int n, a, b, root;
    while(cin>>n)
    {
        memset(pre,0,sizeof(pre));
        memset(dp,0,sizeof(dp));
        for(int i=0; i<=n; i++) tree[i].clear();

        for(int i=1; i<=n; i++) scanf("%d",&w[i]);
        while(scanf("%d%d",&a,&b),a+b)
        {
            tree[b].push_back(a);
            pre[a]=b;
        }
        for(int i=1; i<=n; i++) if(pre[i]==0)   root=i;
        DFS(root);
        printf("%d\n", max(dp[root][0],dp[root][1]) );
    }
    return 0;
}