Cod sursa(job #2620403)

Utilizator DariusGhercaDarius Gherca DariusGherca Data 28 mai 2020 20:31:56
Problema Asmax Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <fstream>
#include <iostream>
using namespace std;

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

const int N=16005;
int n,m,lst[N],vf[2*N],urm[2*N],nr=0,val[N],maxi=-1;
bool viz[N];
int s[N];
void adaugare(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)
            {
                val[x]+=val[y];
            }
        }
    }
}
int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        f>>x;
        val[i]=x;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        f>>x>>y;
        adaugare(x,y);
        adaugare(y,x);
    }
    int maxi=-1;
    dfs(1);
    for(int i=1;i<=n;i++)
    if(val[i]>maxi)
    maxi=val[i];
    g<<maxi;
    return 0;
}