Cod sursa(job #2749466)

Utilizator dariadragomir23Dragomir Daria dariadragomir23 Data 6 mai 2021 19:08:41
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f ("asmax.in");
ofstream g ("asmax.out");
bool viz[16001];
int n, mx=INT_MIN, suma[16001], costuri[16001];
vector<int>adiacenta[16001];

void DFS(int nod)
{
    suma[nod]=costuri[nod];
    for(int i=0; i<adiacenta[nod].size(); i++)
    {
        int vecin=adiacenta[nod][i];
        if(!viz[vecin])
        {
            viz[vecin]=1;
            DFS(vecin);
            if(suma[vecin]>=0)
                suma[nod]+=suma[vecin];
        }
    }
}
int main()
{
    f>>n;
    for(int i=1; i<=n; i++)
        f>>costuri[i];
    for(int i=1; i<n; i++)
    {
        int x, y;
        f>>x>>y;
        adiacenta[x].push_back(y);
        adiacenta[y].push_back(x);
    }
    viz[1]=1;
    DFS(1);
    for(int i=1; i<=n; i++)
        mx=max(mx, suma[i]);
    g<<mx;
}