Cod sursa(job #1442941)

Utilizator pincucatalinPincu Catalin pincucatalin Data 26 mai 2015 16:11:26
Problema Asmax Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <cstdio>

using namespace std;
FILE *in=fopen ("asmax.in","r");
FILE *out=fopen ("asmax.out","w");
const int N=16000;
int n,vf[2*N],lst[N],urm[2*N],m,a[N],suma[N],s;
bool viz[N];
void adauga (int x,int y)
{
    vf[++m]=y;
    urm[m]=lst[x];
    lst[x]=m;
}
int rezolvare (int x)
{
    int p,y;
    viz[x]=true;
    suma[x]=a[x];
    p=lst[x];
    while (p!=0)
    {
        y=vf[p];
        if (!viz[y])
        {
            //viz[y]=true;
            rezolvare(y);
            if (suma[y]+suma[x]>suma[x])
                suma[x]+=suma[y];
        }
        p=urm[p];
    }
}
void citire ()
{
    fscanf (in,"%d",&n);
    for (int i=1;i<=n;i++)
    fscanf (in,"%d",&a[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;
        fscanf (in,"%d%d",&x,&y);
        adauga (x,y);
        adauga (y,x);
    }
    rezolvare(1);
    int s=-999999999;
    for (int i=1;i<=n;i++)
        if (s<suma[i]) s=suma[i];
    fprintf (out,"%d",s);

}
int main()
{
    citire();
    return 0;
}