HDU1520：Anniversary party（树形dp第一发）

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

一个公司去参加宴会，要求去的人不能有直接领导关系，给出每一个人的欢乐值，和L K代表K是L的直接领导，问最大的欢乐值是多少。

将公司的关系建为一棵树，从最大的老板向下dfs

dp[i][0] 代表以编号为i的那个员工为根的一棵子树（不包含i）的最大欢乐值。

dp[i][1] 代表以编号为i的那个员工为根的一棵子树（包含i）的最大欢乐值。

那么得到状态转移方程假设j为i的下属

dp[i][0] = max( dp[j][0],dp[j][1] ) ;

dp[i][1] = max( dp[j][0] ) + c[i] ;

 

PS：我对于树形dp根本属于一点不懂的状态，为了做题，先看了题解，之后再补吧。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define mod 100000000
using namespace std;
struct node
{
    int x,y,next;
}eg[1000010];
int n,w[6010],v[6010],head[6010],dp[6010][2],tt;
void init()
{
    tt=0;
    memset(head,-1,sizeof(head));
    memset(v,0,sizeof(v));
    memset(dp,0,sizeof(dp));
}
void add(int xx,int yy)
{
    eg[tt].x=xx;
    eg[tt].y=yy;
    eg[tt].next=head[xx];
    head[xx]=tt++;
    eg[tt].x=yy;
    eg[tt].y=xx;
    eg[tt].next=head[yy];
    head[yy]=tt++;
}
void dfs(int u)
{
    v[u]=1;
    dp[u][1]=w[u];
    for(int i=head[u];i!=-1;i=eg[i].next)
    {
        int bi=eg[i].y;
        if(v[bi]==0)
        {
            dfs(bi);
            dp[u][0]+=max(dp[bi][0],dp[bi][1]);
            dp[u][1]+=dp[bi][0];
        }
    }
}
int main()
{
    int xx,yy;
    while(scanf("%d",&n)!=EOF)
    {
      for(int i=1;i<=n;i++)
      {
          scanf("%d",&w[i]);
      }
      init();
      while(scanf("%d%d",&xx,&yy)!=EOF)
      {
          if(xx==0&&yy==0) break;
          add(xx,yy);
      }
      dfs(1);
      printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}