Cod sursa(job #2623294)

Utilizator DariusGhercaDarius Gherca DariusGherca Data 2 iunie 2020 21:44:59
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <iostream>
#include <cstring>
#include <fstream>
#include <climits>
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
const int N=16001;
int vf[2*N],lst[N],urm[2*N],nr=0,s[N],n,maxi=INT_MIN;
bool viz[N];
void adauga(int x,int y)
{
    vf[++nr]=y;
    urm[nr]=lst[x];
    lst[x]=nr;
}
void dfs(int x)
{
    viz[x]=true;
    for(int p=lst[x];p!=0;p=urm[p])
    {
        int y=vf[p];
        if(!viz[y])
        {
            dfs(y);
            if(s[y]>0)
            s[x]+=s[y];
        }
    }
    maxi=max(maxi,s[x]);
}
int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
    {
        f>>s[i];
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        f>>x>>y;
        adauga(x,y);
        adauga(y,x);
    }
    dfs(1);
    g<<maxi;
    return 0;
}