285. 没有上司的舞会
题目链接 https://www.acwing.com/problem/content/287/
题目:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N=6010;
int w[N];
vector<int > q[N];
bool fa[N];
int f[N][2];
void dfs(int u){
f[u][1]=w[u];
for(int i=0;i<q[u].size();i++){
int j=q[u][i];
dfs(j);
f[u][1]+=f[j][0];
f[u][0]+=max(f[j][0],f[j][1]);
}
}
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
int x,y;
for(int i=1;i<n;i++){
scanf("%d%d",&x,&y);
q[y].push_back(x);
fa[x]=1;
}
int t=1;
while(fa[t]) t++;
dfs(t);
printf("%d\n",max(f[t][0],f[t][1]));
return 0;
}