题目链接:http://acm.split.hdu.edu.cn/showproblem.php?pid=1520
题意:上司和直系下属不能同时参加party,求party的最大活跃值。
输入: 输入n个结点,接下来的n行,表示1~n的每个结点分别具有的活跃值,输入a,b,表示b是a的上司,当a==0&& b==0时,该组数据输入结束。
输出:由于直接有上司和下属关系的两个人不能同时参加party, 求出能让party活跃值最大的方案(求出最大的活跃值即可).
状态转移方程: dp[i][1]+=dp[j][0];
题意:上司和直系下属不能同时参加party,求party的最大活跃值。
输入: 输入n个结点,接下来的n行,表示1~n的每个结点分别具有的活跃值,输入a,b,表示b是a的上司,当a==0&& b==0时,该组数据输入结束。
输出:由于直接有上司和下属关系的两个人不能同时参加party, 求出能让party活跃值最大的方案(求出最大的活跃值即可).
状态转移方程: dp[i][1]+=dp[j][0];
dp[i][0]+=max(dp[j][0],dp[j][0]);
其中i是j的父亲结点。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=; struct node
{
int v;
int next;
} node[N<<]; int dp[N][];
int a[N],head[N],in[N];
int k; void addedge(int u,int v) ///u是父亲,v是孩子
{
node[k].v=v;
node[k].next=head[u];
head[u]=k++;
} int dfs(int u)
{
dp[u][]=;
dp[u][]=a[u];
for(int i=head[u]; i!=-; i=node[i].next)
{
int v=node[i].v;
dfs(v);
dp[u][]+=max(dp[v][],dp[v][]);
dp[u][]+=dp[v][];
}
return max(dp[u][],dp[u][]);
} int main()
{
int n,x,y;
while(scanf("%d",&n)==)
{
k=;
memset(head,-,sizeof(head));
memset(dp,,sizeof(dp));
memset(in,,sizeof(in));
for(int i=; i<=n; i++)
scanf("%d",&a[i]);
while(scanf("%d%d",&x,&y) && (x+y))
{
addedge(y,x);
in[x]++;
}
int ans=;
for(int i=; i<=n; i++) ///其实这是一个森林,找到每棵树的根,求和
if(!in[i]) ans+=dfs(i);
printf("%d\n",ans);
}
return ;
}