AcWing 285. 没有上司的舞会
原题链接
简单
作者:
未名湖畔的梦
,
2023-11-18 11:27:03
,
所有人可见
,
阅读 36
#include <iostream>
#include <cstdio>
using namespace std;
int dp[2][6010];
int f[2][6010];
int ind[6010];
int vis[6010];
int root;
void dfs(int u){
if(!u) return;
vis[u]=1;
root=u;
dp[0][f[0][u]]+=max(dp[1][u]+f[1][u],dp[0][u]);
dp[1][f[0][u]]+=dp[0][u];
ind[f[0][u]]--;
if(!ind[f[0][u]]) dfs(f[0][u]);
}
int main(){
int n=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&f[1][i]);
int a,b;
for(int i=1;i<n;i++){
scanf("%d%d",&a,&b);
f[0][a]=b;
ind[b]++;
}
for(int i=1;i<=n;i++)
if(!vis[i]&&!ind[i])
dfs(i);
printf("%d\n",max(dp[0][root],dp[1][root]+f[1][root]));
return 0;
}