fork download 
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll long long
  5. ll cost[1000000];
  6.  
  7. void dfs(int node,vector<int>t[],vector<int>&v,vector<int>&parent,vector<int>&b){
  8.     v[node]=1;
  9.     for(auto it:t[node]){
  10.         if(v[it]==0){
  11.             parent[it]=node;
  12.             dfs(it,t,v,parent,b);
  13.         }
  14.     }
  15.     ll c1=0;
  16.     for(auto child:t[node]){
  17.         if(child!=parent[node])c1=max(c1,cost[child]);
  18.     }
  19.     cost[node]=c1+b[node];
  20. }
  21.  
  22. int main(){
  23.     ios_base::sync_with_stdio(false);
  24.     cin.tie(NULL);
  25.     int n;
  26.     cin>>n;
  27.     vector<int>t[n+1],b(n+1);
  28.     for(int i=1;i<=n;i++)cin>>b[i];
  29.     for(int i=1;i<n;i++){
  30.         int x,y;
  31.         cin>>x>>y;
  32.         t[x].push_back(y);
  33.         t[y].push_back(x);
  34.     }
  35.     vector<int>v(n+1),parent(n+1);
  36.     dfs(1,t,v,parent,b);
  37.     ll c=-10000000;
  38.     for(int i=1;i<=n;i++)c=max(c,cost[i]);
  39.     cout<<c<<"\n";
  40.     return 0;
  41. }
Success #stdin #stdout 0s 5324KB
comments (?)
stdin 
5
5 7 -10 4 15 
1 2
2 3 
3 4 
1 5
stdout 
20
https://ideone.com/9ObPeV
language:
C++14 (gcc 8.3)
created:
7 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!