Cod sursa(job #1093036)

Utilizator Juve45UAIC Alexandru Ionita Juve45 Data 27 ianuarie 2014 18:17:37
Problema Asmax Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <fstream>
#include <vector>
using namespace std;
#define dmax 16001
vector <int > v[dmax];
int n, use[dmax], val[dmax];
int smax=-9999999;


ifstream fin("asmax.in");
ofstream fout("asmax.out");

void read()
{
    int a, b;
    fin>>n;
    for(int i=1; i<=n; i++)
        fin>>val[i];
    for(int i=1; i<n; i++)
    {
        fin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
}

int cut_tree(int vf)
{
    int k=0;
    use[vf]=1;

        for(int i=0; i<v[vf].size(); i++)
        {
            int y=v[vf][i];
            if(use[y]==0)
                    k=k+cut_tree(y);
        }
if(k>smax)
    smax=k;

        if(k+val[vf]>0)
            return k+val[vf];
            else return 0;

}

int main()
{
    read();

cut_tree(1);

fout<<smax<<'\n';
    return 0;
}