Mai intai trebuie sa te autentifici.

Cod sursa(job #2835401)

Utilizator LionMan101Achim-Panescu Silvian LionMan101 Data 18 ianuarie 2022 17:46:13
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.22 kb
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
const int INF=1e9+7;
const ll INFF=1e18+7;
#define nl '\n'
void __print(int x) { cout << x; }
void __print(long x) { cout << x; }
void __print(long long x) { cout << x; }
void __print(unsigned x) { cout << x; }
void __print(unsigned long x) { cout << x; }
void __print(unsigned long long x) { cout << x; }
void __print(float x) { cout << x; }
void __print(double x) { cout << x; }
void __print(long double x) { cout << x; }
void __print(char x) { cout << '\'' << x << '\''; }
void __print(const char* x) { cout << '\"' << x << '\"'; }
void __print(const string& x) { cout << '\"' << x << '\"'; }
void __print(bool x) { cout << (x ? "true" : "false"); }
template<typename T, typename V>
void __print(const pair<T, V>& x) { cout << '{'; __print(x.first); cout << ','; __print(x.second); cout << '}'; }
template<typename T>
void __print(const T& x) { int f = 0; cout << '{'; for (auto& i : x) cout << (f++ ? "," : ""), __print(i); cout << "}"; }
void _print() { cout << "]\n"; }
template <typename T, typename... V>
void _print(T t, V... v) { __print(t); if (sizeof...(v)) cout << ", "; _print(v...); }
#ifndef ONLINE_JUDGE
#define debug(x...) cout << "[" << #x << "] = ["; _print(x)
#else
#define debug(x...)
#endif

vector<int> gr(16001);
vector<vector<int>> a(16001);
int n,maxx=-INF;

void dfs(int x, int p){
    for(auto&i:a[x]){
        if(i==p) continue;
        dfs(i,x);
        if(gr[i]>0){
            gr[x]+=gr[i];
        } 
    }
    maxx=max(maxx,gr[x]);
}

void solve()
{
    cin>>n;
    for(int i=0; i<n; i++){
        cin>>gr[i];
    }
    for(int i=1; i<n; i++){
        int x,y;
        cin>>x>>y;
        a[--x].emplace_back(--y);
        a[y].emplace_back(x);
    }
    dfs(0,-1);
    cout<<maxx<<nl;
}

int main()
{
// #ifndef ONLINE_JUDGE
    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
// #endif
    freopen("asmax.in", "r", stdin);
    freopen("asmax.out", "w", stdout);
    ios::sync_with_stdio(0);
    cin.tie(0);
    int t=1;
    // int t;
    // cin >> t;
    for(int tt=1; tt<=t; tt++){
        // cout<<"#Case "<<t<<nl;
        solve();
    }
    return 0;
}