版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/cnyali/article/details/77816016
/*
【题意】
公司有n个人,每个人有价值vi,有一天举办年会,每个人都可以参加,但有严格的等级制度,参加活动时,不能同时出现a和a的上司,问如何才能使总和最大。
【分析】
每个人只有去和不去两种状态,设f[i][0]和f[i][1]分别表示第i个人不参加和参加年会,获得的总的最大价值。
则状态转移方程为:
f[i][1] += f[j][0],
f[i][0] += max{f[j][0],f[j][1]};
其中j为i的孩子节点。
这样,从根节点root进行dfs,最后结果为max{f[root][0],f[root][1]}。
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
const int maxn=6000+5;
using namespace std;
struct node{
int fa;
vector<int>ch;
}a[maxn];
int f[maxn][2],p[maxn];
void dfs(int x){
p[x]=1;
int s=a[x].ch.size();
for(int i=0;i<s;i++){
int k=a[x].ch[i];
if(!p[k]){
dfs(k);
f[x][1]+=f[k][0];
f[x][0]+=max(f[k][0],f[k][1]);
}
}
}
int main(){
freopen("test.in","r",stdin);
freopen("test.out","w",stdout);
int i,j,k,m,n,x,y,root=1;
cin>>n;
for(i=1;i<=n;i++)scanf("%d",&f[i][1]);
while(scanf("%d%d",&x,&y),x+y>0){
a[x].fa=y; a[y].ch.push_back(x);
}
for(i=1;i<=n;i++)
if(a[i].fa==0){root=i;break;}
dfs(root);
printf("%d\n",max(f[root][0],f[root][1]));
return 0;
}