Cod sursa(job #2480279)

Utilizator divianegoescuDivia Negoescu divianegoescu Data 25 octombrie 2019 11:09:38
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 kb
#include <fstream>
#include <vector>
#define K 16005
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector <int> v[K];
int n,i,j,x,y,val[K],best[K],f[K],sol=-1000000000;
void dfs(int x){
    f[x]=1;
    best[x]=val[x];
    for(int i=0;i<v[x].size();i++){
        int fiu=v[x][i];
        if(!f[fiu]){
            dfs(fiu);
            if(best[fiu]>0)
                best[x]+=best[fiu];
        }
    }
    sol=max(sol,best[x]);
}
int main(){
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>val[i];
    for(i=1;i<n;i++){
        fin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1);
    fout<<sol;
    return 0;
}