Cod sursa(job #2050547)

Utilizator GrasuneAlexandruGrasune Alexandru Mihai GrasuneAlexandru Data 28 octombrie 2017 10:23:50
Problema Asmax Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

const int maxN=16050;
int val[maxN],n,tata[maxN],maxi,sm=-999999999;
int din[maxN]; /// suma max a unui subarbore cu radacina in dodul i
vector<int> graf[maxN];
void dfs(int nod)
{
   din[nod]=val[nod];
    for(int i=0; i<graf[nod].size(); i++)
    {
        int vecin=graf[nod][i];
        if(tata[nod]!=vecin)
        {
            tata[vecin]=nod;
            dfs(vecin);
            if(din[vecin]>0)
            {
                din[nod]+=din[vecin];

            }
         sm=max(sm,din[nod]);
        }


    }
}

int main()

{
    ifstream fin("asmax.in");
    ofstream fout("asmax.out");
    fin>>n;
    for(int i=1; i<=n; i++)
        fin>>val[i];
    for(int i=1; i<n; i++)
    {
        int x,y;
        fin>>x>>y;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    dfs(1);

   fout<<sm;

    return 0;
}