Cod sursa(job #2486115)

Utilizator catalingudiu@yahoo.roGudiu Catalin [email protected] Data 2 noiembrie 2019 12:14:00
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.94 kb

#include<bits/stdc++.h>
#define maxn 16005
using namespace std;

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

int n,maxim=INT_MIN,maxx=INT_MIN;
vector<int>lista[maxn];
int val[maxn],dp[maxn];

void citire()
{
    fin>>n;
    for(int i=1;i<=n;i++){
        fin>>val[i];
        if(val[i]>maxx)
            maxx=val[i];
    }
    int x,y;
    for(int i=1;i<n;i++)
    {
        fin>>x>>y;
        lista[x].push_back(y);
        lista[y].push_back(x);
    }
}

void dfs(int node,int tata)
{
    for(auto x: lista[node])
        if(x!=tata)
            dfs(x,node);

    dp[node]=val[node];
    for(auto x:lista[node])
    {
        if(x==tata) continue;
        dp[node]+=dp[x];
    }
    dp[node]=max(dp[node],0);
}

int main()
{
    citire();
    dfs(1,1);
    for(int i=1;i<=n;i++)
        maxim=max(maxim,dp[i]);
    if(maxim==0)
        fout<<maxx;
    else
        fout<<maxim;
    return 0;
}