hdu1520 Anniversary party 简单树形DP

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

思路：树形DP的入门题

定义dp[root][1]表示以root为根节点的子树，且root本身参加party的最优值，那么dp[root][1]+=Σdp[son][0]+happy[root];

dp[root][0]表示以root为跟节点且root本身没有参加的最优值，那么dp[root][0]+=max(dp[son][0],dp[son][1]);

如果不理解，可以参考我对hdu1054的解释。

代码如下：

 #include<iostream>
 #include<cstdlib>
 #include<cstdio>
 #include<cstring>
 using namespace std;
 #define MAX 6010
 int dp[MAX][];
 int happy[MAX];
 class node
 {
   public:
   int to;
   int next;
 };
 node edge[*MAX];
 int head[MAX];
 int vis[MAX];
 int n;
 int tol;
 void init()
 {
   memset(vis,,sizeof(vis));
   memset(head,-,sizeof(head));
   memset(dp,,sizeof(dp));
   tol=;
 }
 void Build_Tree(int u,int v)
 {
         edge[tol].to=v;
         edge[tol].next=head[u];
         head[u]=tol++;
 }
 void dfs(int root)
 {
    vis[root]=;
    int hy=happy[root];
    for(int i=head[root];i!=-;i=edge[i].next)
    {
      if(vis[edge[i].to]) continue;
      int son=edge[i].to;
      dfs(son);

      dp[root][]+=dp[son][];
      dp[root][]+=max(dp[son][],dp[son][]);
    }
    dp[root][]+=hy;
 }
 int main()
 {
    while(scanf("%d",&n)!=EOF)
    {
      init();
      for(int i=;i<=n;i++)
        scanf("%d",&happy[i]);
        int u,v;

        while(scanf("%d%d",&v,&u)!=EOF&&v&&u)
        {
         Build_Tree(u,v);
         Build_Tree(v,u);
        }
     dfs();
     cout<<max(dp[][],dp[][])<<endl;
    }
   return ;
  }