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的孩子)。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<vector> 6 using namespace std; 7 #define MAXN 6060 8 9 int dp[MAXN][2]; 10 int n,ans; 11 vector<vector<int> >G; 12 13 void dfs(int u,int father) 14 { 15 for(int i=0;i<G[u].size();i++){ 16 int v=G[u][i]; 17 if(v==father)continue; 18 dfs(v,u); 19 dp[u][0]+=max(dp[v][0],dp[v][1]); 20 dp[u][1]+=dp[v][0]; 21 } 22 } 23 24 int main() 25 { 26 int u,v; 27 while(~scanf("%d",&n)){ 28 G.clear(); 29 G.resize(n+2); 30 memset(dp,0,sizeof(dp)); 31 for(int i=1;i<=n;i++){ 32 scanf("%d",&dp[i][1]); 33 } 34 while(~scanf("%d%d",&u,&v)){ 35 if(u==0&&v==0)break; 36 G[u].push_back(v); 37 G[v].push_back(u); 38 } 39 dfs(1,-1); 40 printf("%d\n",max(dp[1][0],dp[1][1])); 41 } 42 return 0; 43 }