Cod sursa(job #2244558)

Utilizator UnseenMarksmanDavid Catalin UnseenMarksman Data 23 septembrie 2018 01:10:53
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <algorithm>
#include <vector>
#include <climits>
#define maxn 16002
using namespace std;

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

int n, v[maxn], sol[maxn], res=INT_MIN;
bool viz[maxn];
vector<int>A[maxn];

void dfs(int nod)
{
    viz[nod]=true;
    sol[nod]=sol[nod]+v[nod];
    for(auto it:A[nod])
    {
        if(!viz[it])
        {
            dfs(it);
            if(sol[it]>0)
            {
                sol[nod]=sol[nod]+sol[it];
            }
        }
    }
}

int main()
{
    int x,y;
    fin>>n;
    for(int i=1; i<=n; i++)
    {
        fin>>v[i];
    }
    for(int i=1; i<n; i++)
    {
        fin>>x>>y;
        A[x].push_back(y);
        A[y].push_back(x);
    }
    dfs(1);
    for(int i=1; i<=n; i++) res=max(res,sol[i]);
    fout<<res<<'\n';
    return 0;
}