Cod sursa(job #2745221)

Utilizator VanillaSoltan Marian Vanilla Data 26 aprilie 2021 08:49:44
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.77 kb
#include <bits/stdc++.h>
#include <fstream>
using namespace std;
typedef long long int64;
typedef vector<int> vec;
typedef vector<int64> vec64;
    // string __fname = "";
    // ifstream in (__fname + ".in");
    // ofstream out (__fname + ".out");
    // #define cin in
    // #define cout out
#define ss cout << " ";
#define nn cout << "\n";
#define ct(x) cout << x;
#define cts(x) cout << x << " ";
#define ctn(x) cout << x << "\n";
#define db(x) cout << "> " << #x << ": " << x << "\n";
#define qr queries();
void solve(int);
void YN(bool b){if (b){ctn("YES");}else{ctn ("NO");}};
void yn(bool b){if (b){ctn("Yes");}else{ctn ("No");}};
void queries(){int n;cin >> n;for (int i = 1; i <= n; i++) solve(i);}
int64 ceildiv(int64 a, int64 b) {return a / b + !!(a % b);}
// // // // // // // // // // // // // // // // // // // // // // 
/*                  TEMPLATE - VANILLA                         */
// // // // // // // // // // // // // // // // // // // // // //
const int maxn = 16002;
const int64 mod = 1000000007;
const double pi = 3.14159265359;
const int dx[] = {-1, 0, 1, 0};
const int dy[] = {0, 1, 0, -1};

int n;
int rs = -1e9;
vector <int> v[maxn];
vector <int> a(maxn, 0);
vector <int> viz(maxn);
 
void solve(int id){

    return;
}

int dfs (int node) {
    viz[node] = 1;
    int s = a[node];
    for (int i = 0; i < v[node].size(); i++){
        int next = v[node][i];
        if (!viz[next]) {
            s = max(s, s + dfs(next));
        }
    }
    rs = max (rs, s);
    return s;
}
 
 
int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i];
    for (int i = 0; i < n-1; i++) {
        int x,y;
        cin >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1);
    ctn(rs);
    return 0;
}