Cod sursa(job #2499533)

Utilizator anamariatoaderAna Toader anamariatoader Data 26 noiembrie 2019 11:59:44
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 kb
#include <fstream>
#include <vector>

using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");

int n,i,x,y,Max,v[16005],d[16005];
bool viz[16005];
vector <int> g[16005];

void dfs(int nod){
    viz[nod]=1;
    d[nod]=v[nod];
    for(int i=0;i<g[nod].size();i++){
        int nou=g[nod][i];
        if(!viz[nou]){
            dfs(nou);
            d[nod]=max(d[nod],d[nod]+d[nou]);
        }
    }
    Max=max(Max,d[nod]);
}

int main(){
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>v[i];
    for(i=1;i<n;i++){
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    Max=-20000000;
    dfs(1);
    fout<<Max;
    return 0;
}