Cod sursa(job #2623269)

Utilizator DariusGhercaDarius Gherca DariusGherca Data 2 iunie 2020 21:09:16
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <iostream>
#include <cstring>
#include <fstream>
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;
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];
        }
    }
}
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);
    }
    for(int i=1;i<=n;i++)
    {
        if(!viz[i])
        dfs(i);
    }
    int maxi=-1;
    for(int i=1;i<=n;i++)
    {
        if(s[i]>maxi)
        maxi=s[i];
    }
    g<<maxi;
    return 0;
}