Cod sursa(job #2519999)

Utilizator alexdumitrescuDumitrescu George Alex alexdumitrescu Data 8 ianuarie 2020 19:12:03
Problema Asmax Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.68 kb
#include <bits/stdc++.h>
#define Nmax 16005

using namespace std;
ifstream fin ("asmax.in");
ofstream fout ("asmax.out");
int n, s[Nmax], viz[Nmax], maxx, x, y;
vector <int> v[Nmax];
void dfs(int x)
{
    viz[x]=1;
    for(int i=0;i<v[x].size();i++)
        if(viz[v[x][i]]==0)
        {
            dfs(v[x][i]);
            s[x]=max(s[i], s[v[x][i]]+s[x]);
        }
}
int main()
{
    fin >> n;
    for(int i=1;i<=n;i++)
        fin >> s[i];
    for(int i=1;i<n;i++)
    {
        fin >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1);
    for(int i=1;i<=n;i++)
        maxx=max(maxx, s[i]);
    fout << maxx;
    return 0;
}