POJ 2342 && HDU 1520 Anniversary party（树形DP）

Description

话说一个公司的一些然要去参加一个party，每个人有一个愉悦值，而如果某个人的直接上司在场的话会非常扫兴，所以避免这样的安排，问给出n个人，每个人的愉悦值以及他们的上司所属关系，问你让那些人去可以让总的愉悦值最大，并求出这个值

Input

第一行为总人数n，之后n行为每个人的愉悦值，然后每行两个整数a和b表示所属关系b是a的上司，以0 0结束所属关系输入

Output

输出最大总愉悦值

Sample Input

7

1

1

1

1

1

1

1

1 3

2 3

6 4

7 4

4 5

3 5

0 0

Sample Output

5

Solution

这个公司的人事关系可以根据给出的数据得到一个树，最上面的是最高层，往下依次，我们要做的就是在树的节点处进行dp。

用dp[i][0]表示当前i这个人不去，dp[i][1]表示当前i这个人去

那么dp[fa[i]][1]+=dp[i][0]

而 dp[fa[i]][0]+=max(dp[i][0],dp[i][1])

Code

#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define maxn 6666
int a[maxn],fa[maxn],dp[maxn][2];
vector<int> vec[maxn];
int n;
void dfs(int root)
{
dp[root][1]=a[root];
int len=vec[root].size();
for(int i=0;i<len;i++)
dfs(vec[root][i]);
for(int i=0;i<len;i++)
{
dp[root][0]+=max(dp[vec[root][i]][1],dp[vec[root][i]][0]);//这个人不去
dp[root][1]+=dp[vec[root][i]][0];//这个人去
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
//初始化
vec[i].clear();
fa[i]=-1;
dp[i][0]=dp[i][1]=0;
}
int u,v;
while(scanf("%d%d",&u,&v),u||v)
{
vec[v].push_back(u);
fa[u]=v;
}
int temp=1;
while(fa[temp]!=-1)//找到根节点
temp=fa[temp];
dfs(temp);//从根节点开始搜索
int ans=max(dp[temp][0],dp[temp][1]);
printf("%d\n",ans);
}
return 0;
}