Cod sursa(job #2347218)

Utilizator ApolodorTudor Fernea Apolodor Data 18 februarie 2019 16:49:15
Problema Asmax Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream fi("asmax.in");
ofstream fo("asmax.out");

int n,x,y,ma=-100000000;
int d[16005],viz[16005];
vector <int> a[16005];

void dfs(int k)
{
    viz[k]=1;
    for(int i=0; i<a[k].size(); i++)
    {
        x=a[k][i];
        if(viz[x]==0)
        {
            dfs(x);
            if(d[x]>0)
                d[k]+=d[x];
        }
    }
}

int main()
{
    fi>>n;
    for(int i=1; i<=n; i++)
    {
        fi>>x;
        d[i]=x;
    }
    for(int i=1; i<=n-1; i++)
    {
        fi>>x>>y;
        a[x].push_back(y);
        a[y].push_back(x);
    }

    dfs(1);

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

    fo<<ma;

}