hdu 1520

树形dp的基础题;

状态转移很简单:老爸没选,儿子可选可不选,最大就行;

老爸选了,儿子肯定不能选;

dp[root][0]+=max(dp[son][1],dp[son][0]);
dp[root][1]+=dp[son][0];

代码:

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<vector>
 4 #define maxn 6003
 5 using namespace std;
 6 
 7 int w[maxn],dp[maxn][2];
 8 vector<int>ve[maxn];
 9 bool vis[maxn];
10 
11 void dfs(int a)
12 {
13     vis[a]=1;
14     dp[a][1]=w[a];
15     dp[a][0]=0;
16     int l=ve[a].size();
17     for(int i=0; i<l; i++)
18     {
19         int v=ve[a][i];
20         if(vis[v]) continue;
21         dfs(v);
22         dp[a][0]+=max(dp[v][1],dp[v][0]);
23         dp[a][1]+=dp[v][0];
24     }
25 }
26 
27 int main()
28 {
29     int n,a,b;
30     while(scanf("%d",&n)!=EOF)
31     {
32         memset(vis,0,sizeof vis);
33         memset(dp,0,sizeof dp);
34         for(int i=1; i<=n; i++)
35         {
36             scanf("%d",&w[i]);
37             ve[i].clear();
38         }
39         while(scanf("%d%d",&a,&b)&&(a+b))
40         {
41             ve[a].push_back(b);
42             ve[b].push_back(a);
43         }
44         dfs(1);
45         printf("%d\n",max(dp[1][0],dp[1][1]));
46     }
47     return 0;
48 }
View Code

 

posted @ 2013-09-17 09:09  Yours1103  阅读(196)  评论(0编辑  收藏  举报