hdu1520 （树形dp）

hdu1520 http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意是给定一棵树，每个结点有一个价值，要我们选择任意个结点使得总价值最大，规则是如果父亲结点被选了，那么儿子结点不可以被选，但是儿子的儿子可以被选

本来学搜索的时候找到这题搜索题，然后用搜索做的

 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
 #include <algorithm>
 #include <iostream>
 #include <queue>
 #include <stack>
 #include <vector>
 #include <map>
 #include <set>
 #include <string>
 #include <math.h>
 using namespace std;
 typedef long long LL;
 const int INF = <<;
 int val[];
 int f[];
 vector<int> g[];
 int dfs(int u, bool flag)//flag标记父亲结点是不是取
 {
     int t,t2;
     if(!flag)
         t = val[u];
     t2 = ;
     for(int i=; i<g[u].size(); ++i)
     {
         int v = g[u][i];
         if(!flag)
         {
             t += dfs(v,true);
         }
         t2 += dfs(v,false);
     }
     if(flag)
         return t2;
     return max(t,t2);
 }
 int main()
 {
     int n,i,x,y;
     while(scanf("%d",&n)!=EOF)
     {

         for(i=; i<=n; ++i)
         {
             g[i].clear();
             scanf("%d",&val[i]);
             f[i] = -;
         }
         while(scanf("%d%d",&x,&y),x)
         {
             g[y].push_back(x);
             f[x] = y;
         }
         x = ;
         while(f[x]!=-)
             x = f[x];
         printf("%d\n",dfs(x,false));
     }
     return ;
 }

但是学树形dp的时候又找到这道题，然后用树形dp重新做了一次，提高上去，运行的时间变少了

dp[u][1] 表示选择结点u可以获得的最大价值

dp[u][0]表示不选择结点u可以获得的最大价值

状态转移方程是 dp[u][1] = max( dp[u][1], dp[u][1]+dp[v][0] ); dp[u][0] = max( dp[u][0],max(dp[u][0]+dp[v][1], dp[u][0]+dp[v][0]) );

 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
 #include <algorithm>
 #include <iostream>
 #include <queue>
 #include <stack>
 #include <vector>
 #include <map>
 #include <set>
 #include <string>
 #include <math.h>
 using namespace std;
 typedef long long LL;
 const int INF = <<;
 const int N =  + ;
 int val[N];
 int dp[N][];//dp[u][1]表示选这个结点的最大值， dp[u][0] 表示不选这个结点的最大值
 struct Edge
 {
     int v,next;
 }g[N<<];
 int head[N],e;

 void dfs(int u, int fa)
 {
     dp[u][] = val[u];
     dp[u][] = ;
     for(int i=head[u]; i!=-; i=g[i].next)
     {
         int v = g[i].v;
         if(v==fa) continue;
         dfs(v,u);
         dp[u][] = max(dp[u][],dp[u][]+dp[v][]);
         dp[u][] = max(dp[u][],max(dp[u][]+dp[v][],dp[u][]+dp[v][] ));

         //dp[u][1] += dp[v][0];
         //dp[u][0] += dp[v][1];
     }
 }
 void init(int n)
 {
     for(int i=; i<=n; ++i)
         head[i] = -;
     e = ;
 }
 void addEdge(int u, int v)
 {
     g[e].v = v;
     g[e].next = head[u];
     head[u] = e++;
 }

 int main()
 {
     int n,i,a,b;
     while(scanf("%d",&n)!=EOF)
     {
         init(n);
         for(i=; i<=n; ++i)
             scanf("%d",&val[i]);
         while(scanf("%d%d",&a,&b),a)
         {
             addEdge(a,b);
             addEdge(b,a);
             //g[a].push_back(b);
             //g[b].push_back(a);
         }
         dfs(,-);
         printf("%d\n",max(dp[][],dp[][]));
     }
     return ;
 }

最近老发现用vector建图提交老师runtime error，