Cod sursa(job #2329192)

Utilizator cosceexcosceex cosceex Data 26 ianuarie 2019 14:21:39
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
#define nmax 1000001
using namespace std;

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

int v[nmax];
bool viz[nmax];
int n,a,b;
vector <int> mat[nmax];
int ras;

void dfs(int x)
{
    viz[x]=true;
    for (int i=0 , m=mat[x].size(); i<m; i++)
    {
        int y=mat[x][i];
        if (viz[y]==true)
            continue;
            dfs(y);
            if (v[y]>=0)
                v[x]+=v[y];
    }
   // ras=max(ras,v[x]);
}

int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
        f>>v[i];
    for(int i=2;i<=n;i++)
    {
        f>>a>>b;
        mat[a].push_back(b);
        mat[b].push_back(a);
    }

    dfs(1);
    ras=v[1];
    for(int i=2;i<=n;i++)
    ras=max(ras, v[i]);

    g<<ras;
    return 0;
}