1319 蓝桥舞会

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+9;
int dp[N][2],a[N],n;//存储快乐值
vector<int>e[N];

void dfs(int x)
{
  //返回条件
  if(!e[x].size()) dp[x][1]=a[x];
 
  for(auto & y :e[x])
  {
    dfs(y); //对x的所有子节点进行遍历
    dp[x][0]+=max(dp[y][1],dp[y][0]);
    dp[x][1]=dp[y][0]+a[x];
  }
}

int main()
{
  cin>>n;
  //输入每位员工的快乐值
  for(int i=1;i<=n;i++)
  {
    cin>>a[i];
  }
  //初始化邻接表
  for(int i=1;i<n;i++)
  {
    int u,v;cin>>u>>v;
    e[v].push_back(u);//将u添加到v当中去，表示节点关系
  }
  dfs(1);
  cout<<max(dp[1][1],dp[1][0])<<endl;
  return 0;
}