Cod sursa(job #2663975)

Utilizator metallidethantralayerIon Cojocaru metallidethantralayer Data 27 octombrie 2020 18:25:05
Problema Asmax Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.62 kb
#include <bits/stdc++.h>

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

int n,v[16005],smax=-1e9,s=-1;
bitset <16005> viz;
vector <int> V[16005];
void Dfs1(int x)
{
    viz[x]=true;
    if(s<0)
        s=0;
    s+=v[x];
    smax=max(smax,s);
    for(int i=0; i<V[x].size(); i++)
        if(!viz[V[x][i]])
            Dfs1(V[x][i]);
}
int main()
{
    f>>n;
    for(int i=1; i<=n; i++)
        f>>v[i];
    for(int i=1; i<n; i++)
    {
        int x,y;
        f>>x>>y;
        V[x].push_back(y);
        V[y].push_back(x);
    }
    Dfs1(1);
    g<<smax;
    return 0;
}