Cod sursa(job #2913675)

Utilizator biancalautaruBianca Lautaru biancalautaru Data 15 iulie 2022 22:45:57
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
#include <vector>
#define INF 16000001
#define DIM 16001
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n,x,y,sol,v[DIM],s[DIM];
vector<int> l[DIM];
bool viz[DIM];

void dfs(int nod) {
    viz[nod]=1;
    s[nod]=v[nod];
    for (int i=0;i<l[nod].size();i++) {
        int fiu=l[nod][i];
        if (viz[fiu]==0) {
            dfs(fiu);
            if (s[fiu]>0)
                s[nod]+=s[fiu];
        }
    }
    sol=max(sol,s[nod]);
}

int main() {
    fin>>n;
    sol=-INF;
    for (int i=1;i<=n;i++) {
        fin>>v[i];
        sol=max(sol,v[i]);
    }
    for (int i=1;i<n;i++) {
        fin>>x>>y;
        l[x].push_back(y);
        l[y].push_back(x);
    }
    dfs(1);
    fout<<sol;
    return 0;
}