Cod sursa(job #3175301)

Utilizator catalinmarincatalinmarin catalinmarin Data 25 noiembrie 2023 16:47:30
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector<int> vecini[16005];
int valori[16005];
int best[16005];
int ans = -1e9;
void dfs(int nod, int tata){
    best[nod] = valori[nod];
    for(int fiu: vecini[nod]){
        if(fiu == tata)
            continue;
        dfs(fiu, nod);
        if (best[fiu] > 0)
            best[nod] += best[fiu];
    }
    ans = max(ans, best[nod]);
}
int main(){
    int n;
    fin >> n;
    for (int i = 1; i <= n; i++){
        fin >> valori[i];
    }
    for (int i = 1; i < n; i++) {
        int x, y;
        fin >> x >> y;
        vecini[x].push_back(y);
        vecini[y].push_back(x);
    }
    dfs(1, 0);
    fout << ans;
    return 0;
}