hdu 1520(简单树形dp）

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

思路：dp[u][0]表示不取u的最大价值，dp[u][1]表示取u的最大价值，于是有dp[u][0]+=max(dp[v][0],dp[v][1]),dp[u][1]+=dp[v][0](其中v是u的孩子)。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 using namespace std;
 #define MAXN 6060

 int dp[MAXN][];
 int n,ans;
 vector<vector<int> >G;

 void dfs(int u,int father)
 {
     for(int i=;i<G[u].size();i++){
         int v=G[u][i];
         if(v==father)continue;
         dfs(v,u);
         dp[u][]+=max(dp[v][],dp[v][]);
         dp[u][]+=dp[v][];
     }
 }

 int main()
 {
     int u,v;
     while(~scanf("%d",&n)){
         G.clear();
         G.resize(n+);
         memset(dp,,sizeof(dp));
         for(int i=;i<=n;i++){
             scanf("%d",&dp[i][]);
         }
         while(~scanf("%d%d",&u,&v)){
             if(u==&&v==)break;
             G[u].push_back(v);
             G[v].push_back(u);
         }
         dfs(,-);
         printf("%d\n",max(dp[][],dp[][]));
     }
     return ;
 }