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=;
 int w[N], dp[N][], pre[N];
 vector<int> tree[N];

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

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

         for(int i=; 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=; i<=n; i++) if(pre[i]==)   root=i;
         DFS(root);
         printf("%d\n", max(dp[root][],dp[root][]) );
     }
     return ;
 }

AC代码