Cod sursa(job #1759278)

Utilizator PaulCbnCiobanu Paul PaulCbn Data 18 septembrie 2016 19:29:22
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <cstdio>
#include <vector>
#include <iostream>

using namespace std;

const int NMAX = 16001;
const int inf = 2000000000;
vector<int> graf[NMAX];
int viz[NMAX],dp[NMAX];
int N,rez=-inf;

void citire()
{
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
        scanf("%d",&dp[i]);
    for(int i=1;i<N;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        graf[a].push_back(b);
        graf[b].push_back(a);
    }

}

void dfs(int curent,int prec)
{
    viz[curent]=1;
    for(vector<int>::iterator ii = graf[curent].begin();ii!=graf[curent].end();ii++)
        if(!viz[*ii])
            dfs(*ii,curent);

    for(vector<int>::iterator ii = graf[curent].begin();ii!=graf[curent].end();ii++)
        if(dp[*ii]>0 && *ii != prec)
            dp[curent]+=dp[*ii];

    rez = max(rez,dp[curent]);
}


int main()
{
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    citire();
    dfs(1,0);
    printf("%d",rez);
    return 0;
}