树形dp第一弹：没有上司的舞会

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
int a[6001],f[6001][2];
vector<int>b[6001];
int boss[6001];
void dfs(int x){
for(int j=0;j<b[x].size();j++){
int y=b[x][j];
dfs(y);
f[x][0]+=max(f[y][0],f[y][1]);
f[x][1]+=f[y][0];
}
f[x][1]+=a[x];
}
int main()
{
int n,i,j,k;
cin>>n;
for(i=1;i<=n;i++)
cin>>a[i];
for(i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
boss[x]=y;
b[y].push_back(x);
}
cin>>j>>k;
for(i=1;i<=n;i++)
if(boss[i]==0)
{
dfs(i);
cout<<max(f[i][0],f[i][1])<<endl;
return 0;
}
}