Cod sursa(job #2562102)

Utilizator radugnnGone Radu Mihnea radugnn Data 29 februarie 2020 12:17:35
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <bits/stdc++.h>
#define DIM 16010
using namespace std;
ifstream fin("f.in");
ofstream fout("f.out");
vector <int> L[DIM];
int f[DIM],val[DIM],d[DIM];
int n,i,a,b,sol;
void dfs(int nod){
    f[nod]=1;
    for(int i=0;i<L[nod].size();i++){
        int vecin=L[nod][i];
        if(!f[vecin]){
            dfs(vecin);
            if(d[vecin]>0)
                d[nod]+=d[vecin];
            sol=max(sol,d[nod]);
        }
    }
}
int main(){
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>d[i];
    for(i=1;i<n;i++){
        fin>>a>>b;
        L[a].push_back(b);
        L[b].push_back(a);
    }
    dfs(1);
    fout<<sol;
    return 0;
}