poj2342 Anniversary party 没有上司的舞会（树形DP基础题）

题目链接：http://poj.org/problem?id=2342

题目大意：某大学举行周年会，有些人要参加，每个人都有个活跃度，但是不能让一个人和他的直接上司同时参加，已知这些人的直接上司，求最大的活跃度。

解析：这些人之间的关系可以构成一棵树，本题是一个基础的树形dp。用dp[i][0]表示以当前结点为根的子树且不选择当前点的最大值，dp[i][1]表示以当前结点为根且选择当前结点的最大值。那么转移方程为：dp[i][0] = sum(max(dp[j][1],dp[j][0]))（j是i的孩子结点），dp[i][1] = sum(dp[j][0])+a[i]。

因为他们之间的关系是一棵树，所以树形dp基本都是按照递归进行求值，本题使用dfs进行递归。具体代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int maxn=7000;
struct node
{
    int nex,to;
}edge[maxn*2];
int n;
int a[maxn];
int bos[maxn];
int dp[maxn][3];
int head[maxn*2];
int cnt;
void add(int x,int y)
{
    edge[cnt].to=y;
    edge[cnt].nex=head[x];
    head[x]=cnt++;
    edge[cnt].to=x;
    edge[cnt].nex=head[y];
    head[y]=cnt++;
}
void dfs(int key,int parent)
{
    for(int i=head[key];i!=-1;i=edge[i].nex)
    {
        int v=edge[i].to;
        if(v==parent)continue;
        dfs(v,key);
        dp[key][0]+=max(dp[v][0],dp[v][1]);
        dp[key][1]+=dp[v][0];
    }
    dp[key][1]+=a[key];
}
int main()
{
    while(~scanf("%d",&n))
    {
       for(int i=1;i<=n;i++)scanf("%d",&a[i]);
       int x,y;
       cnt=0;
       memset(dp,0,sizeof dp);
       memset(head,-1,sizeof head);
       while(~scanf("%d%d",&x,&y)&&x&&y)
       {
           add(x,y);
       }
       dfs(1,0);
       printf("%d\n",max(dp[1][1],dp[1][0]));
    }
    return 0;
}