Cod sursa(job #2768474)

Utilizator gruhtenZinnenberg Gruhten gruhten Data 10 august 2021 22:46:32
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <bits/stdc++.h>
using namespace std;
const int dim=16002;

ifstream f ("asmax.in");
ofstream g ("asmax.out");

int n;
vector <int > G[dim];
int d[dim];
int maxim=-100000;

void dfs(int nod , int parinte)
{
    for(int i=0 ; i<G[nod].size(); ++i)
    {
        if(G[nod][i]!=parinte)
        {
            dfs(G[nod][i],nod);

            if(d[G[nod][i]]>0)
            {
                d[nod]+=d[G[nod][i]];
            }
        }
    }
}

int main()
{
    f>>n;
    for(int i=1; i<=n; ++i)
        f>>d[i];

    for(int i=1; i<n; ++i)
    {

        int x , y;
        f>>x>>y;

        G[x].push_back(y);
        G[y].push_back(x);
    }

    dfs(1,1);

    for(int i=1; i<=n; ++i)
        maxim=max(maxim , d[i]);

    g<<maxim;

    return 0;

}